Skip to content

Commit

Permalink
TST: Update test for windows builds, remove try statement for geotiff. (
Browse files Browse the repository at this point in the history
#1660)

* TST: Update test for windows builds.

* MNT: Remove unused parameter.

* FIX: Update with statement.
  • Loading branch information
zssherman authored Oct 8, 2024
1 parent d21f61f commit 3516606
Showing 1 changed file with 28 additions and 43 deletions.
71 changes: 28 additions & 43 deletions tests/io/test_output_to_geotiff.py
Original file line number Diff line number Diff line change
@@ -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

Expand Down Expand Up @@ -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

0 comments on commit 3516606

Please sign in to comment.