Fix bug with Raster.show #407. Update test accordingly.

GlacioHack · Oct 20, 2023 · f991515 · f991515
1 parent a63b04c
commit f991515
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/geoutils/raster/raster.py b/geoutils/raster/raster.py
@@ -2514,12 +2514,7 @@ def show(
 
         # Create axes
         if ax is None:
-            # If no figure exists, get a new axis
-            if len(plt.get_fignums()) == 0:
-                ax0 = plt.gca()
-            # Otherwise, get first axis
-            else:
-                ax0 = plt.gcf().axes[0]
+            ax0 = plt.gca()
         elif isinstance(ax, str) and ax.lower() == "new":
             _, ax0 = plt.subplots()
         elif isinstance(ax, matplotlib.axes.Axes):

diff --git a/tests/test_raster.py b/tests/test_raster.py
@@ -2147,6 +2147,14 @@ def test_show(self) -> None:
         img_RGB = gu.Raster(self.landsat_rgb_path)
 
         # Test default plot
+        img.show()
+        if DO_PLOT:
+            plt.show()
+        else:
+            plt.close()
+        assert True
+
+        # Test with provided ax
         ax = plt.subplot(111)
         img.show(ax=ax, title="Simple plotting test")
         if DO_PLOT: