mirror of
https://github.com/rasterio/rasterio.git
synced 2025-12-08 17:36:12 +00:00
55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
"""Unittests for rasterio.mask"""
|
|
|
|
|
|
import pytest
|
|
|
|
import rasterio
|
|
from rasterio.mask import mask as mask_tool
|
|
|
|
|
|
def test_nodata(basic_image_file, basic_geometry):
|
|
nodata_val = 0
|
|
geometries = [basic_geometry]
|
|
with rasterio.open(basic_image_file, "r") as src:
|
|
masked, transform = mask_tool(src, geometries, crop=False,
|
|
nodata=nodata_val, invert=True)
|
|
assert(masked.data.all() == nodata_val)
|
|
|
|
|
|
def test_no_nodata(basic_image_file, basic_geometry):
|
|
default_nodata_val = 0
|
|
geometries = [basic_geometry]
|
|
with rasterio.open(basic_image_file, "r") as src:
|
|
masked, transform = mask_tool(src, geometries, crop=False, invert=True)
|
|
assert(masked.data.all() == default_nodata_val)
|
|
|
|
|
|
def test_crop(basic_image, basic_image_file, basic_geometry):
|
|
geometries = [basic_geometry]
|
|
with rasterio.open(basic_image_file, "r") as src:
|
|
masked, transform = mask_tool(src, geometries, crop=True)
|
|
|
|
image = basic_image
|
|
image[4, :] = 0
|
|
image[:, 4] = 0
|
|
assert(masked.shape == (1, 4, 3))
|
|
assert((masked[0] == image[1:5, 2:5]).all())
|
|
|
|
|
|
def test_crop_all_touched(basic_image, basic_image_file, basic_geometry):
|
|
geometries = [basic_geometry]
|
|
with rasterio.open(basic_image_file, "r") as src:
|
|
masked, transform = mask_tool(src, geometries, crop=True,
|
|
all_touched=True)
|
|
|
|
assert(masked.shape == (1, 4, 3))
|
|
assert((masked[0] == basic_image[1:5, 2:5]).all())
|
|
|
|
|
|
def test_crop_and_invert(basic_image_file, basic_geometry):
|
|
geometries = [basic_geometry]
|
|
with rasterio.open(basic_image_file) as src:
|
|
with pytest.raises(ValueError):
|
|
masked, transform = mask_tool(src, geometries,
|
|
crop=True, invert=True)
|