def test_crop_nothing_to_crop(): arr = np.array( [[0, 4, 0, 3], [0, 4, 4, 3], [0, 1, 1, 3], [0, 1, 1, 3], [0, 0, 0, 0]], dtype=np.int64) raster = create_test_arr(arr) result = crop(raster, raster, zones_ids=(0, )) assert result.shape == arr.shape compare = arr == result.data assert compare.all()
def test_crop(): arr = np.array( [[0, 4, 0, 3], [0, 4, 4, 3], [0, 1, 1, 3], [0, 1, 1, 3], [0, 0, 0, 0]], dtype=np.int64) raster = create_test_arr(arr) result = crop(raster, raster, zones_ids=(1, 3)) assert result.shape == (4, 3) trimmed_arr = np.array([[4, 0, 3], [4, 4, 3], [1, 1, 3], [1, 1, 3]], dtype=np.int64) compare = trimmed_arr == result.data assert compare.all()
def test_crop(): arr = np.array( [[0, 4, 0, 3], [0, 4, 4, 3], [0, 1, 1, 3], [0, 1, 1, 3], [0, 0, 0, 0]], dtype=np.int64) raster = create_test_arr(arr) # add crs for tests raster = _add_EPSG4326_crs_to_da(raster) result = crop(raster, raster, zones_ids=(1, 3)) assert result.shape == (4, 3) # crs tests assert result.attrs == raster.attrs for coord in raster.coords: assert np.all(result[coord] == raster[coord]) trimmed_arr = np.array([[4, 0, 3], [4, 4, 3], [1, 1, 3], [1, 1, 3]], dtype=np.int64) compare = trimmed_arr == result.data assert compare.all()