import pytest

from app.algorithms.leakage.identifier import LeakageIdentifier


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


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