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