OrgTJWater/TJWaterServer
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add reservoir test case

Browse Source
This commit is contained in:
wqy
2022-09-17 19:38:43 +08:00
parent 89ffd15b7b
commit b8d2efdc66
6 changed files with 57 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
test_tjnetwork.py
View File

@@ -36,7 +36,7 @@ class TestApi:
change_set = add_junction(p, "j0", 0.0, 10.0, 20.0)
assert len(change_set.added) == 1
assert change_set.added[0]['type'] == 'junction'
assert change_set.added[0]['type'] == JUNCTION
assert change_set.added[0]['id'] == "j0"
coord = get_junction_coord(p, 'j0')
assert coord['x'] == 0.0
@@ -48,16 +48,18 @@ class TestApi:
assert get_junction_demand(p, 'j0') == "NULL"
change_set = set_junction_demand(p, 'j0', 100.0)
assert len(change_set.updated) == 1
assert change_set.updated[0]['type'] == 'junction'
assert change_set.updated[0]['type'] == JUNCTION
assert change_set.updated[0]['id'] == 'j0'
assert change_set.updated[0]['property'] == 'demand'
assert change_set.updated[0]['property_type'] == 'float'
assert change_set.updated[0]['property_value'] == str(100.0)
assert get_junction_demand(p, 'j0') == 100.0
# TODO: pattern
change_set = add_junction(p, "j1", 0.0, 10.0, 20.0)
assert len(change_set.added) == 1
assert change_set.added[0]['type'] == 'junction'
assert change_set.added[0]['type'] == JUNCTION
assert change_set.added[0]['id'] == "j1"
nodes = get_nodes(p)
@@ -67,7 +69,7 @@ class TestApi:
change_set = delete_junction(p, 'j1')
assert len(change_set.deleted) == 1
assert change_set.deleted[0]['type'] == 'junction'
assert change_set.deleted[0]['type'] == JUNCTION
assert change_set.deleted[0]['id'] == "j1"
nodes = get_nodes(p)
@@ -76,7 +78,7 @@ class TestApi:
change_set = delete_junction(p, 'j0')
assert len(change_set.deleted) == 1
assert change_set.deleted[0]['type'] == 'junction'
assert change_set.deleted[0]['type'] == JUNCTION
assert change_set.deleted[0]['id'] == "j0"
nodes = get_nodes(p)
@@ -84,10 +86,39 @@ class TestApi:
self.leave(p)
pass
def test_reservoir(self):
pass
p = "test_reservoir"
self.enter(p)
change_set = add_reservoir(p, "r0", 0.0, 10.0, 20.0)
assert len(change_set.added) == 1
assert change_set.added[0]['type'] == RESERVOIR
assert change_set.added[0]['id'] == "r0"
assert get_reservoir_head(p, 'r0') == 20.0
change_set = set_reservoir_head(p, 'r0', 100.0)
assert change_set.updated[0]['type'] == RESERVOIR
assert change_set.updated[0]['id'] == 'r0'
assert change_set.updated[0]['property'] == 'head'
assert change_set.updated[0]['property_type'] == 'float'
assert change_set.updated[0]['property_value'] == str(100.0)
assert get_reservoir_head(p, 'r0') == 100.0
assert get_reservoir_coord(p, 'r0') == { 'x': 0.0, 'y': 10.0 }
change_set = set_reservoir_coord(p, 'r0', 100.0, 200.0)
assert change_set.updated[0]['type'] == RESERVOIR
assert change_set.updated[0]['id'] == 'r0'
assert change_set.updated[0]['property'] == 'coord'
assert change_set.updated[0]['property_type'] == 'point'
assert change_set.updated[0]['property_value'] == str({'x': 100.0, 'y': 200.0})
assert get_reservoir_coord(p, 'r0') == { 'x': 100.0, 'y': 200.0 }
change_set = delete_reservoir(p, 'r0')
assert len(change_set.deleted) == 1
assert change_set.deleted[0]['type'] == RESERVOIR
assert change_set.deleted[0]['id'] == "r0"
self.leave(p)
def test_tank(self):
pass