mirror of
https://github.com/rasterio/rasterio.git
synced 2025-12-08 17:36:12 +00:00
09b6ac6d29
GDALAutoCreateWarpedVRT handles most of the setup (as it wraps GDALCreateWarpedVRT), so some of it needs manual initialization.
104 lines
3.8 KiB
Python
104 lines
3.8 KiB
Python
import boto3
|
|
from packaging.version import parse
|
|
import pytest
|
|
|
|
import rasterio
|
|
from rasterio.enums import Resampling
|
|
from rasterio.transform import Affine
|
|
from rasterio.vrt import WarpedVRT
|
|
from rasterio.windows import Window
|
|
|
|
|
|
# Custom markers.
|
|
mingdalversion = pytest.mark.skipif(
|
|
parse(rasterio.__gdal_version__) < parse('2.1.0dev'),
|
|
reason="S3 raster access requires GDAL 2.1")
|
|
|
|
credentials = pytest.mark.skipif(
|
|
not(boto3.Session()._session.get_credentials()),
|
|
reason="S3 raster access requires credentials")
|
|
|
|
|
|
def test_warped_vrt(path_rgb_byte_tif):
|
|
"""A VirtualVRT has the expected VRT properties."""
|
|
with rasterio.open(path_rgb_byte_tif) as src:
|
|
vrt = WarpedVRT(src, dst_crs='EPSG:3857')
|
|
assert vrt.dst_crs == 'EPSG:3857'
|
|
assert vrt.src_nodata is None
|
|
assert vrt.dst_nodata is None
|
|
assert vrt.tolerance == 0.125
|
|
assert vrt.resampling == Resampling.nearest
|
|
assert vrt.warp_extras == {'init_dest': 'NO_DATA'}
|
|
|
|
|
|
def test_warped_vrt_source(path_rgb_byte_tif):
|
|
"""A VirtualVRT has the expected source dataset."""
|
|
with rasterio.open(path_rgb_byte_tif) as src:
|
|
vrt = WarpedVRT(src, dst_crs='EPSG:3857')
|
|
assert vrt.src_dataset == src
|
|
|
|
|
|
def test_wrap_file(path_rgb_byte_tif):
|
|
"""A VirtualVRT has the expected dataset properties."""
|
|
with rasterio.open(path_rgb_byte_tif) as src:
|
|
vrt = WarpedVRT(src, dst_crs='EPSG:3857')
|
|
assert vrt.crs == 'EPSG:3857'
|
|
assert tuple(round(x, 1) for x in vrt.bounds) == (
|
|
-8789636.7, 2700460.0, -8524406.4, 2943560.2)
|
|
assert vrt.name.startswith('WarpedVRT(')
|
|
assert vrt.name.endswith('tests/data/RGB.byte.tif)')
|
|
assert vrt.indexes == (1, 2, 3)
|
|
assert vrt.nodatavals == (0, 0, 0)
|
|
assert vrt.dtypes == ('uint8', 'uint8', 'uint8')
|
|
assert vrt.read().shape == (3, 736, 803)
|
|
|
|
|
|
def test_warped_vrt_dimensions(path_rgb_byte_tif):
|
|
"""
|
|
A WarpedVRT with target dimensions has the expected dataset
|
|
properties.
|
|
"""
|
|
with rasterio.open(path_rgb_byte_tif) as src:
|
|
extent = (-20037508.34, 20037508.34)
|
|
size = (2 ** 16) * 256
|
|
resolution = (extent[1] - extent[0]) / size
|
|
dst_transform = Affine(resolution, 0.0, extent[0],
|
|
0.0, -resolution, extent[1])
|
|
vrt = WarpedVRT(src, dst_crs='EPSG:3857',
|
|
dst_width=size, dst_height=size,
|
|
dst_transform=dst_transform)
|
|
assert vrt.dst_crs == 'EPSG:3857'
|
|
assert vrt.src_nodata is None
|
|
assert vrt.dst_nodata is None
|
|
assert vrt.resampling == Resampling.nearest
|
|
assert vrt.width == size
|
|
assert vrt.height == size
|
|
assert vrt.transform == dst_transform
|
|
assert vrt.warp_extras == {'init_dest': 'NO_DATA'}
|
|
|
|
|
|
def test_warp_extras(path_rgb_byte_tif):
|
|
"""INIT_DEST warp extra is passed through."""
|
|
with rasterio.open(path_rgb_byte_tif) as src:
|
|
with WarpedVRT(src, dst_crs='EPSG:3857', init_dest=255) as vrt:
|
|
rgb = vrt.read()
|
|
assert (rgb[:, 0, 0] == 255).all()
|
|
|
|
|
|
@mingdalversion
|
|
@credentials
|
|
@pytest.mark.network
|
|
def test_wrap_s3():
|
|
"""A warp wrapper's dataset has the expected properties"""
|
|
L8TIF = "s3://landsat-pds/L8/139/045/LC81390452014295LGN00/LC81390452014295LGN00_B1.TIF"
|
|
with rasterio.open(L8TIF) as src:
|
|
with WarpedVRT(src, dst_crs='EPSG:3857') as vrt:
|
|
assert vrt.crs == 'EPSG:3857'
|
|
assert tuple(round(x, 1) for x in vrt.bounds) == (
|
|
9556764.6, 2345109.3, 9804595.9, 2598509.1)
|
|
assert vrt.name == 'WarpedVRT(s3://landsat-pds/L8/139/045/LC81390452014295LGN00/LC81390452014295LGN00_B1.TIF)'
|
|
assert vrt.indexes == (1,)
|
|
assert vrt.nodatavals == (None,)
|
|
assert vrt.dtypes == ('uint16',)
|
|
assert vrt.shape == (7827, 7655)
|