def test_place_vertical_shift(source, center, vals_request):
# shift 1 cell (10 meters) up
place = raster.Place(source, "EPSG:28992", center, [(50, 60)])
values = place.get_data(**vals_request)["values"]
# swap Y axis for readable test
values = values[:, ::-1, :]
assert (values[:, 1:11, :10] == 7).all()
assert (values[:, 0, :] == 255).all()
def test_place_horizontal_shift(source, center, vals_request):
# shift 1 cell (10 meters) to the right
place = raster.Place(source, "EPSG:28992", center, [(60, 50)])
values = place.get_data(**vals_request)["values"]
# swap Y axis for readable test
values = values[:, ::-1, :]
assert (values[:, :10, 1:11] == 7).all()
assert (values[:, :, 0] == 255).all()
def test_place_reproject(source, center_epsg3857):
target = 572050, 6812050 # EPSG3857 coords somewhere inside RD validity
place = raster.Place(source, "EPSG:3857", center_epsg3857, [target])
x, y = shapely_transform(Point(*target), "EPSG:3857", "EPSG:28992").coords[0]
values = place.get_data(
mode="vals",
bbox=(x - 40, y - 40, x + 40, y + 40),
projection="EPSG:28992",
width=8,
height=8,
)["values"]
assert (values == 7).all()
def test_place_multiple(source, center, vals_request, statistic, expected):
# place such that only the left and bottom ridges have values
place = raster.Place(
source, "EPSG:28992", center, [(-40, 50), (50, -40)], statistic
)
values = place.get_data(**vals_request)["values"]
# swap Y axis for readable test
values = values[:, ::-1, :]
# zero features
assert (values[:, 1:, 1:] == expected[0]).all()
# one feature
assert (values[:, 1:10, 0] == expected[1]).all()
assert (values[:, 0, 1:10] == expected[2]).all()
# two features
assert (values[:, 0, 0] == expected[3]).all()
def test_place_attrs_reproject(source, center_epsg3857):
# Place should adapt the spatial attributes (extent & geometry)
place = raster.Place(
source, "EPSG:3857", center_epsg3857, [(572050, 6812050), (570050, 6811050)]
)
# The native projection != store projection:
assert place.projection is None
assert place.geo_transform is None
extent_epsg3857 = 570000, 6811000, 572100, 6812100
extent_epsg4326 = shapely_transform(
box(*extent_epsg3857), "EPSG:3857", "EPSG:4326"
).bounds
x1, x2, y1, y2 = place.geometry.GetEnvelope()
assert (x1, y1, x2, y2) == pytest.approx(extent_epsg3857, rel=1e-4)
assert place.extent == pytest.approx(extent_epsg4326, rel=1e-4)
def test_place_attrs(source, center):
place = raster.Place(source, "EPSG:28992", center, [(50, 50)])
# Place should not adapt these attributes:
assert place.period == source.period
assert place.timedelta == source.timedelta
assert place.dtype == source.dtype
assert place.fillvalue == source.fillvalue
# If the place projection equals the store projection:
assert place.projection == source.projection
assert place.geo_transform == source.geo_transform
extent_epsg28992 = (0, 0, 100, 100)
extent_epsg4326 = shapely_transform(
box(*extent_epsg28992), "EPSG:28992", "EPSG:4326"
).bounds
x1, x2, y1, y2 = place.geometry.GetEnvelope()
assert (x1, y1, x2, y2) == extent_epsg28992
assert place.extent == pytest.approx(extent_epsg4326, rel=1e-4)
def test_place_point_request(source, center, point, expected):
# For point requests, edges are important. Let's do a drawing:
# 20 _______
# | |
# | 1 | 2
# 10 |_______|_______
# | |
# 3 | 4 |
# 0 |_______|
# 0 10 20
# - zone 1 and 4 are filled; zone 2 and 3 are empty (see below coordinates)
# A pixel includes its topleft corner and top and left edges
# - line between 2-4 and 3-4 are filled; line 1-2 and 1-3 are empty
# - center point at (10, 10) is filled
coordinates = [(60, -40), (-40, 60)]
place = raster.Place(source, "EPSG:28992", anchor=center, coordinates=coordinates)
point_request = dict(
mode="vals", bbox=point * 2, projection="EPSG:28992", width=1, height=1
)
values = place.get_data(**point_request)["values"]
assert values.shape == (1, 1, 1)
assert values.item() == expected
def test_place_invalid_statistic(source, center):
with pytest.raises(ValueError):
raster.Place(source, "EPSG:28992", center, [(50, 50)], statistic="nonexisting")
def test_place_meta_request(source, center):
place = raster.Place(source, "EPSG:28992", center, [(150, 50)])
assert source.get_data(mode="meta") == place.get_data(mode="meta")
def test_place_outside(source, center, vals_request):
x1, y1, x2, y2 = vals_request["bbox"]
coordinates = [(x1 - 50, y1), (x1, y1 - 50), (x2 + 50, y2), (x2, y2 + 50)]
place = raster.Place(source, "EPSG:28992", center, coordinates)
values = place.get_data(**vals_request)["values"]
assert (values[:, :, :] == 255).all()
def test_place_exact(source, center, vals_request):
place = raster.Place(source, "EPSG:28992", center, [(50, 50)])
values = place.get_data(**vals_request)["values"]
# swap Y axis for readable test
values = values[:, ::-1, :]
assert (values[:, :10, :10] == 7).all()
def test_place_no_coords(source, center, vals_request):
place = raster.Place(source, "EPSG:28992", center, [])
values = place.get_data(**vals_request)["values"]
assert (values[:, :10, :10] == source.fillvalue).all()
def test_place_empty(empty, center, vals_request):
place = raster.Place(empty, "EPSG:28992", center, [(50, 50)])
assert place.geometry is None
assert place.extent is None
assert place.get_data(**vals_request) is None
