def test_rasterize_without_bounds(mock_rasterize): gv = GeoVector(Polygon.from_bounds(0, 0, 1, 1)) gv.rasterize(dest_resolution=0.1, fill_value=29, nodata_value=-19) expected_shape = [gv.get_shape(gv.crs)] expected_bounds = gv.envelope.get_shape(gv.crs) mock_rasterize.assert_called_with(expected_shape, gv.crs, expected_bounds, 0.1, 29, -19)
def test_rasterize_with_polygon_bounds(mock_rasterize): gv = GeoVector(Polygon.from_bounds(0, 0, 1, 1)) expected_bounds = Polygon.from_bounds(0.1, 0.1, 2, 2) bounds = expected_bounds gv.rasterize(0.00001, bounds=bounds) expected_shape = [gv.get_shape(gv.crs)] mock_rasterize.assert_called_with(expected_shape, gv.crs, expected_bounds, 0.00001, None, None)
def test_rasterize_with_geovector_bounds(mock_rasterize): gv = GeoVector(Polygon.from_bounds(0, 0, 1, 1)) expected_bounds = Polygon.from_bounds(0.1, 0.1, 2, 2) bounds = GeoVector(expected_bounds, crs=gv.crs) gv.rasterize(0.00001, bounds=bounds) expected_shape = [gv.get_shape(gv.crs)] mock_rasterize.assert_called_with(expected_shape, gv.crs, expected_bounds, 0.00001, fill_value=None, dtype=None)
def test_rasterize_with_crs(mock_rasterize): gv = GeoVector(Polygon.from_bounds(0, 0, 1, 1)) expected_crs = WEB_MERCATOR_CRS expected_bounds = Polygon.from_bounds(11132, 11132, 222639, 222684) bounds = GeoVector(Polygon.from_bounds(0.1, 0.1, 2, 2)) gv.rasterize(1000, bounds=bounds, crs=WEB_MERCATOR_CRS) expected_shape = [gv.get_shape(expected_crs)] args, kwargs = mock_rasterize.call_args assert args[0] == expected_shape assert args[1] == expected_crs assert args[2].equals_exact(expected_bounds, tolerance=0.5)