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()
