Add reservoir test case

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

View File

@@ -189,8 +189,8 @@ def set_junction_coord(name: str, junction_id: str, x: float, y: float) -> Chang
# reservoir 3.[RESERVOIRS]
############################################################
def add_reservoir(name: str, reservoir_id: str, x: float, y: float, elevation: float) -> ChangeSet:
return api.add_reservoir(name, reservoir_id, x, y, elevation)
def add_reservoir(name: str, reservoir_id: str, x: float, y: float, head: float) -> ChangeSet:
return api.add_reservoir(name, reservoir_id, x, y, head)
def delete_reservoir(name: str, reservoir_id: str) -> ChangeSet:
return api.delete_reservoir(name, reservoir_id)
@@ -204,8 +204,8 @@ def get_reservoir_pattern(name: str, reservoir_id: str) -> str | None:
def get_reservoir_coord(name: str, reservoir_id: str) -> dict[str, float] | None:
return api.get_reservoir_coord(name, reservoir_id)
def set_reservoir_head(name: str, reservoir_id: str, elevation: float) -> ChangeSet:
return api.set_reservoir_head(name, reservoir_id, elevation)
def set_reservoir_head(name: str, reservoir_id: str, head: float) -> ChangeSet:
return api.set_reservoir_head(name, reservoir_id, head)
def set_reservoir_pattern(name: str, reservoir_id: str, pattern: str) -> ChangeSet:
return api.set_reservoir_pattern(name, reservoir_id, pattern)