def test_retrieve_data_ls5_mndwi(config=None):
filename = "LS5_TM_MNDWI_{x:03d}_{y:04d}_{date}.{x_offset:04d}_{y_offset:04d}.{x_size:04d}x{y_size:04d}.tif".format(x=CELL_X, y=CELL_Y, date=DATE, x_offset=X_OFFSET, y_offset=Y_OFFSET, x_size=X_SIZE, y_size=Y_SIZE)
tiles = list_tiles_as_list(x=[CELL_X], y=[CELL_Y],
acq_min=ACQ_LS5, acq_max=ACQ_LS5,
satellites=[Satellite.LS5],
dataset_types=[MNDWI_DATASET_TYPE],
config=config)
assert len(tiles) == 1
dataset = tiles[0].datasets[MNDWI_DATASET_TYPE]
data = get_dataset_data(dataset=dataset, x=X_OFFSET, y=Y_OFFSET, x_size=X_SIZE, y_size=Y_SIZE)
assert(data)
_log.info("data is [%s]\n%s", numpy.shape(data), data)
ndv = get_dataset_ndv(dataset)
assert(is_ndv(ndv, MNDWI_NDV))
data_type = get_dataset_datatype(dataset)
assert(data_type == MNDWI_DATA_TYPE)
metadata = generate_dataset_metadata(x=CELL_X, y=CELL_Y, acq_dt=ACQ_LS5,
dataset=dataset, bands=None,
mask_pqa_apply=False, mask_pqa_mask=None,
mask_wofs_apply=False, mask_wofs_mask=None)
raster_create_geotiff(filename, [data[b] for b in dataset.bands], CELL_GEO_TRANSFORM, CELL_PROJECTION, ndv, data_type,
dataset_metadata=metadata, band_ids=[b.name for b in dataset.bands])
assert filecmp.cmp(filename, get_test_data_path(filename))
def test_retrieve_data_ls5_arg_with_pqa_water_mask_dry(config=None):
filename = "LS5_TM_NBAR_WITH_PQA_WATER_DRY_{x:03d}_{y:04d}_{date}.{x_offset:04d}_{y_offset:04d}.{x_size:04d}x{y_size:04d}.tif".format(x=CELL_X, y=CELL_Y, date=DATE, x_offset=X_OFFSET, y_offset=Y_OFFSET, x_size=X_SIZE, y_size=Y_SIZE)
tiles = list_tiles_as_list(x=[CELL_X], y=[CELL_Y],
acq_min=ACQ_LS5, acq_max=ACQ_LS5,
satellites=[Satellite.LS5],
dataset_types=[ARG_DATASET_TYPE, PQ_DATASET_TYPE, WOFS_DATASET_TYPE],
config=config)
assert len(tiles) == 1
tile = tiles[0]
assert ARG_DATASET_TYPE in tile.datasets
dataset = tile.datasets[ARG_DATASET_TYPE]
assert PQ_DATASET_TYPE in tile.datasets
pqa = tile.datasets[PQ_DATASET_TYPE]
assert WOFS_DATASET_TYPE in tile.datasets
wofs = tile.datasets[WOFS_DATASET_TYPE]
mask = get_mask_pqa(pqa, x=X_OFFSET, y=Y_OFFSET, x_size=X_SIZE, y_size=Y_SIZE)
mask = get_mask_wofs(wofs, wofs_masks=[WofsMask.DRY, WofsMask.NO_DATA, WofsMask.SATURATION_CONTIGUITY,
WofsMask.SEA_WATER, WofsMask.TERRAIN_SHADOW, WofsMask.HIGH_SLOPE,
WofsMask.CLOUD_SHADOW, WofsMask.CLOUD],
x=X_OFFSET, y=Y_OFFSET, x_size=X_SIZE, y_size=Y_SIZE, mask=mask)
data = get_dataset_data_masked(dataset=dataset, x=X_OFFSET, y=Y_OFFSET, x_size=X_SIZE, y_size=Y_SIZE, mask=mask)
assert(data)
_log.info("data is [%s]\n%s", numpy.shape(data), data)
ndv = get_dataset_ndv(dataset)
assert(is_ndv(ndv, ARG_NDV))
data_type = get_dataset_datatype(dataset)
assert(data_type == ARG_DATA_TYPE)
metadata = generate_dataset_metadata(x=CELL_X, y=CELL_Y, acq_dt=ACQ_LS5,
dataset=dataset, bands=None,
mask_pqa_apply=False, mask_pqa_mask=None,
mask_wofs_apply=False, mask_wofs_mask=None)
raster_create_geotiff(filename, [data[b] for b in dataset.bands], CELL_GEO_TRANSFORM, CELL_PROJECTION, ndv, data_type,
dataset_metadata=metadata, band_ids=[b.name for b in dataset.bands])
assert filecmp.cmp(filename, get_test_data_path(filename))