comment sections in test_plot.py

modflowpy · Aug 15, 2023 · 5272fd2 · 5272fd2
1 parent d092bd6
commit 5272fd2
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/autotest/test_plot.py b/autotest/test_plot.py
@@ -357,6 +357,9 @@ def test_model_dot_plot_export(function_tmpdir, example_data_path):
             raise AssertionError("Plot filenames not written correctly")
 
 
+# %% plotting MP6 models
+
+
 @pytest.fixture
 def mp6_model(function_tmpdir, example_data_path):
     # test with multi-layer example
@@ -387,6 +390,9 @@ def mp6_model(function_tmpdir, example_data_path):
     return ml, mp, sim
 
 
+# %% plotting MP7 models
+
+
 @requires_exe("mf2005", "mp6")
 def test_plot_map_view_mp6_pathline(mp6_model):
     ml, mp, sim = mp6_model
@@ -518,6 +524,9 @@ def test_plot_map_view_mp6_endpoint(mp6_model):
     assert isinstance(ep, PathCollection)
 
 
+# %% plotting MP7 models
+
+
 @pytest.fixture
 def mp7_model(function_tmpdir, example_data_path):
     pass
@@ -533,6 +542,9 @@ def test_plot_cross_section_mp7_pathline(mp7_model):
     pass
 
 
+# %% plotting PRT models
+
+
 @pytest.fixture
 def prt_model(function_tmpdir, example_data_path):
     pass
@@ -548,6 +560,9 @@ def test_plot_cross_section_prt_pathline(prt_model):
     pass
 
 
+# %% PRT-MP7 conversion functions
+
+
 @requires_exe("mf6")
 def test_to_mp7_pathlines(prt_model):
     pass
@@ -558,6 +573,9 @@ def test_to_mp7_endpoints(prt_model):
     pass
 
 
+# %% quasi-3d model plots
+
+
 @pytest.fixture
 def quasi3d_model(function_tmpdir):
     mf = Modflow("model_mf", model_ws=function_tmpdir, exe_name="mf2005")
@@ -681,6 +699,9 @@ def test_cross_section_with_quasi3d_layers(quasi3d_model):
     plt.close()
 
 
+# %% test cross section line representations
+
+
 def structured_square_grid(side: int = 10, thick: int = 10):
     """
     Creates a basic 1-layer structured grid with the given thickness and number of cells per side