mirror of
https://github.com/rasterio/rasterio.git
synced 2026-01-25 14:10:17 +00:00
12f8fc7e3f
New Env class and tests. More logging, tests, safer environment Remove usage (deprecated) of rasterio.drivers() Also pep8 cleanups throughout the tests and a fix for unchecked dtypes when opening a dataset in 'w' mode. Now we can simply import warnings in __init__.py. In the deprecations tests, we needed to see a single warning only one time to avoid multiple drivers() warnings. Add a global env. Add rasterio.env.setenv()
84 lines
2.2 KiB
Python
84 lines
2.2 KiB
Python
import numpy as np
|
|
|
|
try:
|
|
import matplotlib.pyplot as plt
|
|
except ImportError:
|
|
plt = None
|
|
|
|
import rasterio
|
|
from rasterio.plot import show, show_hist
|
|
from rasterio.rio.insp import stats
|
|
|
|
def test_stats():
|
|
with rasterio.open('tests/data/RGB.byte.tif') as src:
|
|
results = stats((src, 1))
|
|
assert results[0] == 0
|
|
assert results[1] == 255
|
|
assert np.isclose(results[2], 29.9477)
|
|
|
|
results2 = stats(src.read(1))
|
|
assert np.allclose(np.array(results), np.array(results2))
|
|
|
|
|
|
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
|
|
|
|
|
|
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
|
|
|
|
|
|
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
|
|
|
|
|
|
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
|