def test_mapzen_elevation_rgb(): """ Should work as expected """ arr = np.random.randint(0, 3000, size=(512, 512)) assert utils.mapzen_elevation_rgb(arr).shape == (3, 512, 512)
def load_assets(x, y, z, assets, tile_size, input_format: str = None, output_format: str = None, backfill: bool = False, pixel_selection: str = 'first', resampling_method: str = "nearest"): if input_format == 'terrarium': arrays = [rasterio.open(asset).read() for asset in assets] backfilled = backfill_arrays(*arrays) if output_format == 'terrarium': return backfilled data = decode_ele(backfilled, 'terrarium', backfill=backfill) elif input_format == 'geotiff': arrays = [rasterio.open(asset).read() for asset in assets] data = backfill_arrays(*arrays) else: with rasterio.Env(aws_session): pixsel_method = PIXSEL_METHODS[pixel_selection] data, _ = mosaic_tiler( assets, x, y, z, cogeoTiler, tilesize=tile_size, pixel_selection=pixsel_method(), resampling_method=resampling_method, ) if output_format == 'terrarium': return mapzen_elevation_rgb(data) return data