Guard pump

2023-03-22 21:00:58 +08:00
parent 6a304ba7e0
commit da796f0c7f
2 changed files with 19 additions and 1 deletions
--- a/test_tjnetwork.py
+++ b/test_tjnetwork.py
@@ -1156,6 +1156,8 @@ class TestApi:
        assert is_junction(p, 'j3')
        assert is_junction(p, 'j4')

+        assert get_pump(p, 'p0') == {}
+
        add_pump(p, ChangeSet({'id': 'p0', 'node1': 'j1', 'node2': 'j2', 'power': 0.0}))
        p0 = get_pump(p, 'p0')
        assert p0['node1'] == 'j1'
@@ -1191,6 +1193,8 @@ class TestApi:
        links = get_links(p)
        assert len(links) == 0

+        assert get_pump(p, 'p0') == {}
+
        self.leave(p)