import numpy as  np
from shapely.geometry import Polygon
from raster import RasterFunction

raster = RasterFunction(
    xs=np.linspace(0, 1, 11), ys=np.linspace(0, 1, 11),
    data=np.array([np.full((11, 11), 1), np.full((11, 11), 2)])
)
polygon = Polygon([(0.5, 0.5), (0.5, 1), (1, 1), (1, 0.5)])
raster.draw(polygon, 99)

assert np.abs(raster._values[0, 2, 2].item() - 1) < 10e-6
assert np.abs(raster._values[0, 8, 8].item() - 99) < 10e-6
assert np.abs(raster._values[1, 2, 2].item() - 2) < 10e-6
assert np.abs(raster._values[1, 8, 8].item() - 99) < 10e-6