WaterBenchmarkHub/tests/test_ltown.py at main · HammerLabML/WaterBenchmarkHub · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
"""
Module provides tests to test the `water_benchmark_hub.networks.ltown` module.
"""
from epyt_flow.simulation import ScenarioConfig
from water_benchmark_hub import load

from .utils import get_temp_folder


def test_ltown():
    res = load("Network-LTown")

    assert res.get_meta_info() != {}

    # Entire L-Town
    assert isinstance(res.load(download_dir=get_temp_folder()), str)
    assert isinstance(res.load(download_dir=get_temp_folder(), return_scenario=True),
                      ScenarioConfig)

    assert isinstance(res.load(download_dir=get_temp_folder(), use_realistic_demands=True), str)
    assert isinstance(res.load(download_dir=get_temp_folder(), return_scenario=True,
                               use_realistic_demands=True), ScenarioConfig)

    assert isinstance(res.load(download_dir=get_temp_folder(), return_scenario=True,
                               use_realistic_demands=True,
                               include_default_sensor_placement=True), ScenarioConfig)

    # L-Town Area A
    assert isinstance(res.load_ltown_a(download_dir=get_temp_folder()), str)
    assert isinstance(res.load_ltown_a(download_dir=get_temp_folder(), return_scenario=True),
                      ScenarioConfig)

    assert isinstance(res.load_ltown_a(download_dir=get_temp_folder(), use_realistic_demands=True),
                      str)
    assert isinstance(res.load_ltown_a(download_dir=get_temp_folder(), return_scenario=True,
                                       use_realistic_demands=True), ScenarioConfig)

    assert isinstance(res.load_ltown_a(download_dir=get_temp_folder(), return_scenario=True,
                                       use_realistic_demands=True,
                                       include_default_sensor_placement=True), ScenarioConfig)