import pytest

from app.algorithms.dma_leakage_estimation.genetic_optimizer import DmaLeakageOptimizer


@pytest.mark.parametrize(
    ("unit", "expected"),
    [
        ("m3/s", 1.0),
        ("m³/s", 1.0),
        ("m3/h", 3600.0),
        ("m³/h", 3600.0),
    ],
)
def test_dma_leakage_optimizer_accepts_display_flow_units(unit, expected):
    assert DmaLeakageOptimizer._flow_from_m3s(1.0, unit) == expected


def test_dma_leakage_optimizer_accepts_display_flow_units_for_input():
    assert DmaLeakageOptimizer._flow_to_m3s(3600.0, "m³/h") == 1.0