diff --git a/python/ribasim/tests/conftest.py b/python/ribasim/tests/conftest.py index 8efffb016..611fc9c76 100644 --- a/python/ribasim/tests/conftest.py +++ b/python/ribasim/tests/conftest.py @@ -37,3 +37,8 @@ def discrete_control_of_pid_control() -> ribasim.Model: @pytest.fixture() def subnetwork() -> ribasim.Model: return ribasim_testmodels.subnetwork_model() + + +@pytest.fixture() +def main_network_with_subnetworks() -> ribasim.Model: + return ribasim_testmodels.main_network_with_subnetworks_model() diff --git a/python/ribasim/tests/test_model.py b/python/ribasim/tests/test_model.py index d4da03a3b..36998dc6f 100644 --- a/python/ribasim/tests/test_model.py +++ b/python/ribasim/tests/test_model.py @@ -131,8 +131,9 @@ def test_tabulated_rating_curve_model(tabulated_rating_curve, tmp_path): Model.read(tmp_path / "tabulated_rating_curve/ribasim.toml") -def test_plot(discrete_control_of_pid_control): +def test_plot(discrete_control_of_pid_control, main_network_with_subnetworks): discrete_control_of_pid_control.plot() + main_network_with_subnetworks.plot() def test_write_adds_fid_in_tables(basic, tmp_path):