rasterio/tests/test_sampling.py
Sean Gillies 394756330c Workaround for #378.
Adds a pure Python generator to be called by RasterReader.sample.
2015-06-26 14:04:48 -06:00

27 lines
818 B
Python

import rasterio
def test_sampling():
with rasterio.open('tests/data/RGB.byte.tif') as src:
data = next(src.sample([(220650.0, 2719200.0)]))
assert list(data) == [18, 25, 14]
def test_sampling_beyond_bounds():
with rasterio.open('tests/data/RGB.byte.tif') as src:
data = next(src.sample([(-10, 2719200.0)]))
assert list(data) == [0, 0, 0]
def test_sampling_indexes():
with rasterio.open('tests/data/RGB.byte.tif') as src:
data = next(src.sample([(220650.0, 2719200.0)], indexes=[2]))
assert list(data) == [25]
def test_sampling_type():
"""See https://github.com/mapbox/rasterio/issues/378."""
with rasterio.open('tests/data/RGB.byte.tif') as src:
sampler = src.sample([(220650.0, 2719200.0)], indexes=[2])
assert type(sampler)