diff --git a/test_tjnetwork.py b/test_tjnetwork.py index a71d220..344fb5a 100644 --- a/test_tjnetwork.py +++ b/test_tjnetwork.py @@ -612,6 +612,35 @@ class TestApi: self.leave(p) + def test_reservoir_del(self): + p = 'test_reservoir_del' + self.enter(p) + + add_junction(p, ChangeSet({'id': 'j0', 'x': 0.0, 'y': 10.0, 'elevation': 20.0})) + add_reservoir(p, ChangeSet({'id': 'r0', 'x': 0.0, 'y': 10.0, 'head': 20.0})) + add_pipe(p, ChangeSet({'id': 'p0', 'node1': 'j0', 'node2': 'r0', 'length': 100.0, 'diameter': 10.0, 'roughness': 0.1, 'minor_loss': 0.5, 'status': PIPE_STATUS_OPEN })) + assert is_junction(p, 'j0') + assert is_reservoir(p, 'r0') + assert is_pipe(p, 'p0') + + delete_reservoir(p, ChangeSet({'id': 'r0'})) + assert is_junction(p, 'j0') + assert is_reservoir(p, 'r0') == False + assert is_pipe(p, 'p0') == False + + execute_undo(p) + assert is_junction(p, 'j0') + assert is_reservoir(p, 'r0') + assert is_pipe(p, 'p0') + + execute_redo(p) + assert is_junction(p, 'j0') + assert is_reservoir(p, 'r0') == False + assert is_pipe(p, 'p0') == False + + self.leave(p) + + # 4 tank