fix(test_sfr): update test to be more robust with Matplotlib versions

modflowpy · Feb 14, 2023 · e238d2c · e238d2c
1 parent 649596d
commit e238d2c
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/autotest/test_sfr.py b/autotest/test_sfr.py
@@ -797,12 +797,8 @@ def test_sfr_plot(mf2005_model_path):
         check=False,
     )
     sfr = m.get_package("SFR")
-    tv = sfr.plot(
-        key="strtop",
-    )
-    assert issubclass(
-        type(tv[0]), matplotlib.axes.SubplotBase
-    ), "could not plot strtop"
+    tv = sfr.plot(key="strtop")
+    assert isinstance(tv[0], matplotlib.axes.SubplotBase)
 
 
 def get_test_matrix():