archive-gh-me/rasterio
1
0
Fork 0
mirror of https://github.com/rasterio/rasterio.git synced 2025-12-08 17:36:12 +00:00
Code Issues Projects Releases Wiki Activity

add tests

Browse Source
This commit is contained in:
vincentsarago 2023-04-13 21:29:33 +02:00
parent 4de5754e52
commit 9064fabeeb
2 changed files with 44 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
tests/data/RGBA.uint16.tif Normal file
View File

Binary file not shown.

71
tests/test_warpedvrt.py
View File

@ -93,24 +93,25 @@ def test_warped_vrt_add_alpha(dsrec, path_rgb_byte_tif):
def test_warped_vrt_msk_add_alpha(path_rgb_msk_byte_tif, caplog):
"""Add an alpha band to the VRT to access per-dataset mask of a source"""
with rasterio.open(path_rgb_msk_byte_tif) as src:
vrt = WarpedVRT(src, crs=DST_CRS, add_alpha=True)
assert vrt.src_nodata is None
assert vrt.dst_nodata is None
assert vrt.count == 4
assert vrt.mask_flag_enums == (
[MaskFlags.per_dataset, MaskFlags.alpha],
) * 3 + (
[MaskFlags.all_valid],
)
with rasterio.Env(GDAL_DISABLE_READDIR_ON_OPEN="TRUE"):
with rasterio.open(path_rgb_msk_byte_tif) as src:
vrt = WarpedVRT(src, crs=DST_CRS, add_alpha=True)
assert vrt.src_nodata is None
assert vrt.dst_nodata is None
assert vrt.count == 4
assert vrt.mask_flag_enums == (
[MaskFlags.per_dataset, MaskFlags.alpha],
) * 3 + (
[MaskFlags.all_valid],
)
caplog.set_level(logging.DEBUG)
with rasterio.Env(CPL_DEBUG=True):
masks = vrt.read_masks()
assert masks[0, 0, 0] == 0
assert masks[0].mean() > 0
caplog.set_level(logging.DEBUG)
with rasterio.Env(CPL_DEBUG=True):
masks = vrt.read_masks()
assert masks[0, 0, 0] == 0
assert masks[0].mean() > 0
assert "RGB2.byte.tif.msk" in caplog.text
assert "RGB2.byte.tif.msk" in caplog.text
def test_warped_vrt_msk_nodata(path_rgb_msk_byte_tif, caplog):
@ -376,18 +377,19 @@ def test_hit_ovr(red_green):
# GDAL doesn't log overview hits for local files , so we copy the
# overviews of green.tif over the red overviews and expect to find
# green pixels below.
green_ovr = red_green.join("green.tif.ovr")
shutil.move(green_ovr, red_green.join("red.tif.ovr"))
assert not green_ovr.exists()
with rasterio.open(str(red_green.join("red.tif.ovr"))) as ovr:
data = ovr.read()
assert (data[1] == 204).all()
with rasterio.Env(GDAL_DISABLE_READDIR_ON_OPEN="TRUE"):
green_ovr = red_green.join("green.tif.ovr")
shutil.move(green_ovr, red_green.join("red.tif.ovr"))
assert not green_ovr.exists()
with rasterio.open(str(red_green.join("red.tif.ovr"))) as ovr:
data = ovr.read()
assert (data[1] == 204).all()
with rasterio.open(str(red_green.join("red.tif"))) as src, WarpedVRT(src) as vrt:
data = vrt.read(out_shape=(vrt.count, vrt.height // 2, vrt.width // 2))
image = numpy.moveaxis(data, 0, -1)
assert image[0, 0, 0] == 17
assert image[0, 0, 1] == 204
with rasterio.open(str(red_green.join("red.tif"))) as src, WarpedVRT(src) as vrt:
data = vrt.read(out_shape=(vrt.count, vrt.height // 2, vrt.width // 2))
image = numpy.moveaxis(data, 0, -1)
assert image[0, 0, 0] == 17
assert image[0, 0, 1] == 204
def test_warped_vrt_1band_add_alpha():
@ -666,3 +668,18 @@ def test_warpedvrt_rpcs__width_height():
assert vrt.dst_transform.almost_equals(
affine.Affine(0.008598908695300157, 0.0, -123.48824818566573, 0.0, -0.0041566403046337285, 49.52797830474037)
)
def test_warpedvrt_dst_alpha():
"""When using using Uint16 data with Alpha band we should keep the original values"""
with rasterio.open('tests/data/RGBA.uint16.tif') as src:
arr = src.read(1)
assert numpy.unique(arr).tolist() == [10000]
with WarpedVRT(src, add_alpha=False, init_dest=1) as vrt:
vrt_arr = vrt.read(1)
# When using WarpedVRT, the `masked` data will be initialized to the
# `init_dest` value (1) so the unique values in the array should be
# 1 and 10000 (the original unique value).
assert numpy.unique(vrt_arr).tolist() == [1, 10000]