Add direct polygon label QA
This commit is contained in:
@@ -0,0 +1,22 @@
|
||||
from scripts.render_operator_polygon_label_qa import geometry_rings
|
||||
|
||||
|
||||
def test_geometry_rings_yields_polygon_exterior_and_hole() -> None:
|
||||
exterior = [[0, 0], [1, 0], [1, 1], [0, 0]]
|
||||
hole = [[0.2, 0.2], [0.4, 0.2], [0.2, 0.2]]
|
||||
assert list(geometry_rings({"type": "Polygon", "coordinates": [exterior, hole]})) == [
|
||||
exterior,
|
||||
hole,
|
||||
]
|
||||
|
||||
|
||||
def test_geometry_rings_flattens_multipolygon_rings() -> None:
|
||||
first = [[0, 0], [1, 0], [0, 0]]
|
||||
second = [[2, 2], [3, 2], [2, 2]]
|
||||
assert list(
|
||||
geometry_rings({"type": "MultiPolygon", "coordinates": [[first], [second]]})
|
||||
) == [first, second]
|
||||
|
||||
|
||||
def test_geometry_rings_ignores_non_polygon_geometry() -> None:
|
||||
assert list(geometry_rings({"type": "Point", "coordinates": [0, 0]})) == []
|
||||
Reference in New Issue
Block a user