async def test_cog_tiler_point(create_cog_reader):
infile = "https://async-cog-reader-test-data.s3.amazonaws.com/webp_cog.tif"
async with create_cog_reader(infile) as cog:
tiler = COGTiler(cog)
centroid = Polygon.from_bounds(
*transform_bounds(cog.epsg, "EPSG:4326", *cog.bounds)).centroid
val = await tiler.point(coords=[centroid.x, centroid.y],
coords_crs=CRS.from_epsg(4326))
assert list(val) == [50, 69, 74]
async def test_cog_tiler_info(create_cog_reader):
infile = "https://async-cog-reader-test-data.s3.amazonaws.com/webp_cog.tif"
async with create_cog_reader(infile) as cog:
tiler = COGTiler(cog)
info = await tiler.info()
profile = cog.profile
assert info.max_zoom > info.min_zoom
assert info.dtype == profile["dtype"]
assert info.color_interp == profile["photometric"]
async def test_cog_tiler_part(create_cog_reader):
infile_nodata = (
"https://async-cog-reader-test-data.s3.amazonaws.com/naip_image_nodata.tif"
)
async with create_cog_reader(infile_nodata) as cog:
tiler = COGTiler(cog)
arr = await tiler.part(
bounds=(-10526706.9, 4445561.5, -10526084.1, 4446144.0),
bounds_crs=CRS.from_epsg(cog.epsg),
)
assert arr.shape == (3, 976, 1043)
async def test_cog_tiler_tile(create_cog_reader):
infile = "https://async-cog-reader-test-data.s3.amazonaws.com/webp_cog.tif"
async with create_cog_reader(infile) as cog:
tiler = COGTiler(cog)
with rasterio.open(infile) as ds:
_, zoom = get_zooms(ds)
centroid = Polygon.from_bounds(
*transform_bounds(cog.epsg, "EPSG:4326", *cog.bounds)).centroid
tile = mercantile.tile(centroid.x, centroid.y, zoom)
# tile_native_bounds = transform_bounds("EPSG:4326", cog.epsg, *mercantile.bounds(tile))
arr = await tiler.tile(tile.x,
tile.y,
tile.z,
tile_size=256,
resample_method=Image.BILINEAR)
with cogeo_reader(infile) as ds:
rio_tile_arr, rio_tile_mask = ds.tile(tile.x,
tile.y,
tile.z,
tilesize=256,
resampling_method="bilinear")
if cog.is_masked:
tile_arr = np.ma.getdata(arr)
tile_mask = np.ma.getmask(arr)
# Make sure image data is the same
assert pytest.approx(tile_arr - rio_tile_arr,
1) == np.zeros(tile_arr.shape)
# Make sure mask data is the same
rio_mask_counts = np.unique(rio_tile_mask, return_counts=True)
tile_mask_counts = np.unique(tile_mask, return_counts=True)
assert len(rio_mask_counts[0]) == len(tile_mask_counts[0])
assert (rio_mask_counts[1][0] *
cog.profile["count"] == tile_mask_counts[1][0])
else:
assert pytest.approx(arr - rio_tile_arr, 1) == np.zeros(arr.shape)
async def test_cog_tiler_preview_dimensions(create_cog_reader):
infile = "https://async-cog-reader-test-data.s3.amazonaws.com/webp_cog.tif"
async with create_cog_reader(infile) as cog:
tiler = COGTiler(cog)
arr = await tiler.preview(width=512, height=512)
assert arr.shape == (3, 512, 512)
async def test_cog_tiler_preview_max_size(create_cog_reader):
infile = "https://async-cog-reader-test-data.s3.amazonaws.com/webp_cog.tif"
async with create_cog_reader(infile) as cog:
tiler = COGTiler(cog)
arr = await tiler.preview(max_size=512)
assert arr.shape == (3, 512, 432)