Merge branch 'ARM-DOE:main' into update_release_2.0

ARM-DOE · Oct 8, 2024 · 4dbcf89 · 4dbcf89
2 parents a3f8d72 + 3516606
commit 4dbcf89
Showing 1 changed file with 28 additions and 43 deletions.
diff --git a/tests/io/test_output_to_geotiff.py b/tests/io/test_output_to_geotiff.py
@@ -1,11 +1,9 @@
 """ Unit Tests for Py-ART's output_to_geotiff.py module. """
 
 import warnings
-from pathlib import Path
 
 import numpy as np
 import pytest
-from PIL import Image
 
 import pyart
 
@@ -157,51 +155,38 @@ def test_write_grid_geotiff_missing_field():
 )
 def test_write_grid_geotiff_transparent_background():
     grid = make_tiny_grid_with_mask()
-
-    try:
-        with pyart.testing.InTemporaryDirectory() as tmpdir:
-            tmp = Path(tmpdir)
-            outname = tmp / "transparent_bg.tif"
-            pyart.io.write_grid_geotiff(
-                grid,
-                str(outname),
-                "reflectivity",
-                rgb=True,
-                cmap="pyart_HomeyerRainbow",
-                vmin=0,
-                vmax=40,
-                transparent_bg=True,
-                opacity=1,
-            )
-            imgname = outname.rename(tmp / "transparent_bg.tiff")
-            img = Image.open(imgname)
-            img.show()
-    except PermissionError:
-        pass
+    with pyart.testing.InTemporaryDirectory():
+        pyart.io.write_grid_geotiff(
+            grid,
+            "transparent.foo",
+            "reflectivity",
+            rgb=True,
+            cmap="pyart_HomeyerRainbow",
+            vmin=0,
+            vmax=40,
+            transparent_bg=True,
+            opacity=1,
+        )
+        with open("transparent.foo", "rb") as f:
+            assert len(f.read()) > 0
 
 
 @pytest.mark.skipif(
     not pyart.io.output_to_geotiff.IMPORT_FLAG, reason="GDAL is not installed."
 )
 def test_write_grid_geotiff_opacity():
     grid = make_tiny_grid_with_mask()
-    try:
-        with pyart.testing.InTemporaryDirectory() as tmpdir:
-            tmp = Path(tmpdir)
-            outname = tmp / "opacity.tif"
-            pyart.io.write_grid_geotiff(
-                grid,
-                str(outname),
-                "reflectivity",
-                rgb=True,
-                cmap="pyart_HomeyerRainbow",
-                vmin=0,
-                vmax=40,
-                transparent_bg=False,
-                opacity=0.25,
-            )
-            imgname = outname.rename(tmp / "opacity.tiff")
-            img = Image.open(imgname)
-            img.show()
-    except PermissionError:
-        pass
+    with pyart.testing.InTemporaryDirectory():
+        pyart.io.write_grid_geotiff(
+            grid,
+            "opacity.foo",
+            "reflectivity",
+            rgb=True,
+            cmap="pyart_HomeyerRainbow",
+            vmin=0,
+            vmax=40,
+            transparent_bg=False,
+            opacity=0.25,
+        )
+        with open("opacity.foo", "rb") as f:
+            assert len(f.read()) > 0