mirror of
https://github.com/rasterio/rasterio.git
synced 2025-12-08 17:36:12 +00:00
85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
"""Unittests for rasterio.plot"""
|
|
|
|
|
|
import pytest
|
|
|
|
try:
|
|
import matplotlib as mpl
|
|
mpl.use('agg')
|
|
import matplotlib.pyplot as plt
|
|
plt.show = lambda :None
|
|
except ImportError:
|
|
plt = None
|
|
|
|
import rasterio
|
|
from rasterio.plot import show
|
|
from rasterio.plot import show_hist
|
|
|
|
@pytest.mark.skipif(plt is None,
|
|
reason="requires matplotlib")
|
|
def test_show_raster():
|
|
"""
|
|
This test only verifies that code up to the point of plotting with
|
|
matplotlib works correctly. Tests do not exercise matplotlib.
|
|
"""
|
|
with rasterio.open('tests/data/RGB.byte.tif') as src:
|
|
try:
|
|
show((src, 1))
|
|
fig = plt.gcf()
|
|
plt.close(fig)
|
|
except ImportError:
|
|
pass
|
|
|
|
@pytest.mark.skipif(plt is None,
|
|
reason="requires matplotlib")
|
|
def test_show_raster_no_bounds():
|
|
"""
|
|
This test only verifies that code up to the point of plotting with
|
|
matplotlib works correctly. Tests do not exercise matplotlib.
|
|
"""
|
|
with rasterio.open('tests/data/RGB.byte.tif') as src:
|
|
try:
|
|
show((src, 1), with_bounds=False)
|
|
fig = plt.gcf()
|
|
plt.close(fig)
|
|
except ImportError:
|
|
pass
|
|
|
|
|
|
@pytest.mark.skipif(plt is None,
|
|
reason="requires matplotlib")
|
|
def test_show_array():
|
|
"""
|
|
This test only verifies that code up to the point of plotting with
|
|
matplotlib works correctly. Tests do not exercise matplotlib.
|
|
"""
|
|
with rasterio.open('tests/data/RGB.byte.tif') as src:
|
|
try:
|
|
show(src.read(1))
|
|
fig = plt.gcf()
|
|
plt.close(fig)
|
|
except ImportError:
|
|
pass
|
|
|
|
@pytest.mark.skipif(plt is None,
|
|
reason="requires matplotlib")
|
|
def test_show_hist():
|
|
"""
|
|
This test only verifies that code up to the point of plotting with
|
|
matplotlib works correctly. Tests do not exercise matplotlib.
|
|
"""
|
|
with rasterio.open('tests/data/RGB.byte.tif') as src:
|
|
try:
|
|
show_hist((src, 1), bins=256)
|
|
fig = plt.gcf()
|
|
plt.close(fig)
|
|
except ImportError:
|
|
pass
|
|
|
|
try:
|
|
show_hist(src.read(), bins=256)
|
|
fig = plt.gcf()
|
|
plt.close(fig)
|
|
except ImportError:
|
|
pass
|