def test_Raster(self):
polygon = WKTElement('POLYGON((0 0,1 1,0 1,0 0))', srid=4326)
o = Ocean(polygon.ST_AsRaster(5, 5))
session.add(o)
session.flush()
session.expire(o)
assert isinstance(o.rast, RasterElement)
height = session.execute(o.rast.ST_Height()).scalar()
assert height == 5
width = session.execute(o.rast.ST_Width()).scalar()
assert width == 5
# The top left corner is covered by the polygon
top_left_point = WKTElement('Point(0 1)', srid=4326)
top_left = session.execute(o.rast.ST_Value(top_left_point)).scalar()
assert top_left == 1
# The bottom right corner has NODATA
bottom_right_point = WKTElement('Point(1 0)', srid=4326)
bottom_right = session.execute(
o.rast.ST_Value(bottom_right_point)).scalar()
assert bottom_right is None
def test_Raster(self):
polygon = WKTElement('POLYGON((0 0,1 1,0 1,0 0))', srid=4326)
o = Ocean(polygon.ST_AsRaster(5, 5))
session.add(o)
session.flush()
session.expire(o)
assert isinstance(o.rast, RasterElement)
rast_data = (
'01000001009A9999999999C93F9A9999999999C9BF0000000000000000000000000000F03'
'F00000000000000000000000000000000E610000005000500440001010101010101010100'
'010101000001010000000100000000'
)
assert o.rast.data == rast_data
assert session.execute(
select([Ocean.rast.ST_Height(), Ocean.rast.ST_Width()])
).fetchall() == [(5, 5)]
# Set rast to None
o.rast = None
# Insert in DB
session.flush()
session.expire(o)
# Check what was updated in DB
assert o.rast is None
cols = [Ocean.id, Ocean.rast]
if SQLA_LT_2:
assert session.execute(select(cols)).fetchall() == [(1, None)]
else:
assert session.execute(select(*cols)).fetchall() == [(1, None)]
# Reset rast to initial value
o.rast = RasterElement(rast_data)
# Insert in DB
session.flush()
session.expire(o)
# Check what was updated in DB
assert o.rast.data == rast_data
assert session.execute(
select([Ocean.rast.ST_Height(), Ocean.rast.ST_Width()])
).fetchall() == [(5, 5)]
