def test_rasterize_wkt_attrs():
geom = box(135004, 455995, 135004.5, 455996)
view = raster.RasterizeWKT(geom.wkt, "EPSG:28992")
assert view.projection == "EPSG:28992"
assert_almost_equal(view.geometry.GetEnvelope(), [135004, 135004.5, 455995, 455996])
assert view.geometry.GetSpatialReference().IsSame(get_sr("EPSG:28992"))
assert view.dtype == bool
assert view.fillvalue is None
assert_almost_equal(
view.extent, shapely_transform(geom, "EPSG:28992", "EPSG:4326").bounds
)
assert view.timedelta is None
assert view.period == (datetime(1970, 1, 1), datetime(1970, 1, 1))
def test_rasterize_wkt_vals(vals_request, projection):
# vals_request has width=4, height=6 and cell size of 0.5
# we place a rectangle of 2 x 3 with corner at x=1, y=2
view = raster.RasterizeWKT(
shapely_transform(box(135000.5, 455998, 135001.5, 455999.5),
"EPSG:28992", projection).wkt,
projection,
)
vals_request["start"] = vals_request["stop"] = None
actual = view.get_data(**vals_request)
assert actual["values"][0].astype(int).tolist() == [
[0, 0, 0, 0],
[0, 1, 1, 0],
[0, 1, 1, 0],
[0, 1, 1, 0],
[0, 0, 0, 0],
[0, 0, 0, 0],
]
def test_rasterize_wkt_point(point_request, bbox, expected):
view = raster.RasterizeWKT(box(*bbox).wkt, "EPSG:28992")
point_request["start"] = point_request["stop"] = None
actual = view.get_data(**point_request)
assert actual["values"].tolist() == [[[expected]]]
def test_rasterize_wkt_vals_no_intersection(vals_request):
view = raster.RasterizeWKT(box(135004, 455995, 135004.5, 455996).wkt, "EPSG:28992")
vals_request["start"] = vals_request["stop"] = None
actual = view.get_data(**vals_request)
assert ~actual["values"].any()