From a204980944e63d29927e863de4566c71a7930de2 Mon Sep 17 00:00:00 2001
From: Jiang <huarch97@outlook.com>
Date: Fri, 17 Jul 2026 11:45:58 +0800
Subject: [PATCH] fix(leakage): accept display flow units

---
 app/algorithms/leakage/identifier.py  | 16 +++++++++-------
 tests/unit/test_leakage_flow_units.py | 20 ++++++++++++++++++++
 2 files changed, 29 insertions(+), 7 deletions(-)
 create mode 100644 tests/unit/test_leakage_flow_units.py

diff --git a/app/algorithms/leakage/identifier.py b/app/algorithms/leakage/identifier.py
index c2044b1..c02a705 100644
--- a/app/algorithms/leakage/identifier.py
+++ b/app/algorithms/leakage/identifier.py
@@ -121,13 +121,15 @@ def _worker_evaluate(raw_ratios: np.ndarray) -> float:
         _cleanup_temp_files(prefix)
 
 
-class LeakageIdentifier:
-    FLOW_UNIT_TO_M3S = {
-        "m3/s": 1.0,
-        "m3/h": 1.0 / 3600.0,
-        "L/s": 1.0 / 1000.0,
-        "L/min": 1.0 / 60000.0,
-    }
+class LeakageIdentifier:
+    FLOW_UNIT_TO_M3S = {
+        "m3/s": 1.0,
+        "m³/s": 1.0,
+        "m3/h": 1.0 / 3600.0,
+        "m³/h": 1.0 / 3600.0,
+        "L/s": 1.0 / 1000.0,
+        "L/min": 1.0 / 60000.0,
+    }
 
     @classmethod
     def _flow_to_m3s(cls, value: float, unit: str) -> float:
diff --git a/tests/unit/test_leakage_flow_units.py b/tests/unit/test_leakage_flow_units.py
new file mode 100644
index 0000000..bbad2c8
--- /dev/null
+++ b/tests/unit/test_leakage_flow_units.py
@@ -0,0 +1,20 @@
+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