def test_read_all_reaches(file):
q_waterlevel = QueryData("WaterLevel")
ts = read(file, [q_waterlevel])
# Note that it includes 4 water level structure points
assert ts.shape == (110, 247)
q_discharge = QueryData("Discharge")
ts = read(file, [q_discharge])
# Note that it includes 2 discharge structure points
assert ts.shape == (110, 129)
def test_file_does_not_exist():
file = "tests/testdata/not_a_file.res1d"
query = QueryData("WaterLevel")
with pytest.raises(FileExistsError):
assert read(file, [query])
def test_read_multiple_queries(file):
q1 = QueryData("WaterLevel", "104l1", 34.4131)
q2 = QueryData("Discharge", "9l1", 5)
ts = read(file, [q1, q2])
assert ts.shape == (110, 2)
ts_max = ts.max()
assert pytest.approx(round(ts_max[0], 3)) == 197.046
assert pytest.approx(round(ts_max[1], 3)) == 0.761
def test_read_reach(file):
q_reach = QueryData("WaterLevel", "118l1")
ts = read(file, [q_reach])
assert ts.shape == (110, 3)
assert list(ts.columns) == [
'WaterLevel 118l1 0.000',
'WaterLevel 118l1 49.443',
'WaterLevel 118l1 98.887'
]
def test_read_bad_queries(file):
"""Querying data not available in the file must return an error"""
# Bad variable type
with pytest.raises(DataNotFoundInFile) as excinfo:
read(file, [QueryData("Pollutant")])
assert "Pollutant" in str(excinfo.value)
# Bad reach name
with pytest.raises(DataNotFoundInFile) as excinfo:
read(file, [QueryData("WaterLevel", "bad_reach_name")])
assert "bad_reach_name" in str(excinfo.value)
# Bad chainage
with pytest.raises(DataNotFoundInFile) as excinfo:
read(file, [QueryData("WaterLevel", "104l1", 666)])
assert "666" in str(excinfo.value)
def test_read_single_query(file, query, expected_max):
ts = read(file, query)
assert len(ts) == 110
assert pytest.approx(round(ts.max()[0], 3)) == expected_max
def test_read_multiple_reaches(file):
q_reach1 = QueryData("WaterLevel", "118l1")
q_reach2 = QueryData("Discharge", "113l1")
ts = read(file, [q_reach1, q_reach2])
assert ts.shape == (110, 5)