Add valve test case

2022-09-17 23:09:43 +08:00
parent ec3dff12b0
commit 96c01d7fc8
3 changed files with 121 additions and 3 deletions
--- a/tjnetwork.py
+++ b/tjnetwork.py
@@ -369,7 +369,7 @@ def get_valve_node1(name: str, valve_id: str) -> str | None:
    return api.get_valve_node1(name, valve_id)

 def get_valve_node2(name: str, valve_id: str) -> str | None:
-    return api.get_valve_node1(name, valve_id)
+    return api.get_valve_node2(name, valve_id)

 def get_valve_diameter(name: str, valve_id: str) -> float | None:
    return api.get_valve_diameter(name, valve_id)