def test_mask_rasterize_wrong_number_fill(numbers, fill): # _mask_rasterize does not raise on wrong fill numbers or on missing numbers with pytest.raises(ValueError): _mask_rasterize( dummy_lon, dummy_lat, dummy_outlines_poly, numbers=numbers, fill=fill )
def test_rasterize(a, b, fill): expected = expected_mask_2D(a=a, b=b, fill=fill) result = _mask_rasterize( dummy_lon, dummy_lat, dummy_outlines_poly, numbers=[a, b], fill=fill ) assert np.allclose(result, expected, equal_nan=True)
def test_deg45_rasterize_offset_equal(regions): # https://github.com/mathause/regionmask/issues/80 polygons = regions.polygons lon = ds_for_45_deg.lon lat = ds_for_45_deg.lat result_no_offset = _mask_rasterize_no_offset(lon, lat, polygons, numbers=[0]) result_offset = _mask_rasterize(lon, lat, polygons, numbers=[0]) assert np.allclose(result_no_offset, result_offset, equal_nan=True)
def test_mask_rasterize_wrong_number_fill(numbers, fill): with pytest.raises(AssertionError): _mask_rasterize(lon, lat, outlines_poly, numbers=numbers, fill=fill)