Guard reservoir

This commit is contained in:
WQY\qiong
2023-03-22 20:52:47 +08:00
parent e68154ea0b
commit 3afb24c816
2 changed files with 19 additions and 1 deletions

View File

@@ -459,6 +459,8 @@ class TestApi:
p = 'test_reservoir'
self.enter(p)
assert get_reservoir(p, 'r0') == {}
add_reservoir(p, ChangeSet({'id': 'r0', 'x': 0.0, 'y': 10.0, 'head': 20.0}))
r0 = get_reservoir(p, 'r0')
assert r0['x'] == 0.0
@@ -495,6 +497,8 @@ class TestApi:
nodes = get_nodes(p)
assert len(nodes) == 0
assert get_reservoir(p, 'r0') == {}
add_reservoir(p, ChangeSet({'id': 'r1', 'x': 0.0, 'y': 10.0, 'head': 20.0}))
add_reservoir(p, ChangeSet({'id': 'r2', 'x': 0.0, 'y': 10.0, 'head': 20.0}))
add_reservoir(p, ChangeSet({'id': 'r3', 'x': 0.0, 'y': 10.0, 'head': 20.0}))