def test_trim(): arr = np.array( [[0, 0, 0, 0], [0, 4, 0, 0], [0, 4, 4, 0], [0, 1, 1, 0], [0, 0, 0, 0]], dtype=np.int64) raster = create_test_arr(arr) trimmed_raster = trim(raster, values=(0, )) assert trimmed_raster.shape == (3, 2) trimmed_arr = np.array([[4, 0], [4, 4], [1, 1]], dtype=np.int64) compare = trimmed_arr == trimmed_raster.data assert compare.all()
def test_trim_right_top(): arr = np.array( [[0, 0, 0, 0], [4, 0, 3, 0], [4, 4, 3, 0], [1, 1, 3, 0], [1, 1, 3, 0]], dtype=np.int64) raster = create_test_arr(arr) trimmed_raster = trim(raster, values=(0, )) assert trimmed_raster.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 == trimmed_raster.data assert compare.all()
def test_trim(): arr = np.array( [[0, 0, 0, 0], [0, 4, 0, 0], [0, 4, 4, 0], [0, 1, 1, 0], [0, 0, 0, 0]], dtype=np.int64) raster = create_test_arr(arr) # add crs for tests raster = _add_EPSG4326_crs_to_da(raster) trimmed_raster = trim(raster, values=(0, )) assert trimmed_raster.shape == (3, 2) # crs tests assert trimmed_raster.attrs == raster.attrs for coord in raster.coords: assert np.all(trimmed_raster[coord] == raster[coord]) trimmed_arr = np.array([[4, 0], [4, 4], [1, 1]], dtype=np.int64) compare = trimmed_arr == trimmed_raster.data assert compare.all()