Expand plot testing

Deltares · Dec 21, 2023 · ff040aa · ff040aa
1 parent 553c956
commit ff040aa
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git 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
@@ -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):