First Commit

2026-05-31 10:17:09 +07:00
commit 17a9c69379
4547 changed files with 1170384 additions and 0 deletions
@@ -0,0 +1,92 @@
+import unittest
+
+from numpy.testing import assert_array_equal
+
+from shapely.geometry import (
+    GeometryCollection,
+    LinearRing,
+    LineString,
+    MultiLineString,
+    MultiPoint,
+    MultiPolygon,
+    Point,
+    Polygon,
+)
+from shapely.ops import orient
+
+
+class OrientTestCase(unittest.TestCase):
+    def test_point(self):
+        point = Point(0, 0)
+        assert orient(point, 1) == point
+        assert orient(point, -1) == point
+
+    def test_multipoint(self):
+        multipoint = MultiPoint([(0, 0), (1, 1)])
+        assert orient(multipoint, 1) == multipoint
+        assert orient(multipoint, -1) == multipoint
+
+    def test_linestring(self):
+        linestring = LineString([(0, 0), (1, 1)])
+        assert orient(linestring, 1) == linestring
+        assert orient(linestring, -1) == linestring
+
+    def test_multilinestring(self):
+        multilinestring = MultiLineString([[(0, 0), (1, 1)], [(1, 0), (0, 1)]])
+        assert orient(multilinestring, 1) == multilinestring
+        assert orient(multilinestring, -1) == multilinestring
+
+    def test_linearring(self):
+        linearring = LinearRing([(0, 0), (0, 1), (1, 0)])
+        assert orient(linearring, 1) == linearring
+        assert orient(linearring, -1) == linearring
+
+    def test_empty_polygon(self):
+        polygon = Polygon()
+        assert orient(polygon) == polygon
+
+    def test_polygon(self):
+        polygon = Polygon([(0, 0), (0, 1), (1, 0)])
+        polygon_reversed = Polygon(polygon.exterior.coords[::-1])
+        assert (orient(polygon, 1)) == polygon_reversed
+        assert (orient(polygon, -1)) == polygon
+
+    def test_multipolygon(self):
+        polygon1 = Polygon([(0, 0), (0, 1), (1, 0)])
+        polygon2 = Polygon([(1, 0), (2, 0), (2, 1)])
+        polygon1_reversed = Polygon(polygon1.exterior.coords[::-1])
+        polygon2_reversed = Polygon(polygon2.exterior.coords[::-1])
+        multipolygon = MultiPolygon([polygon1, polygon2])
+        assert not polygon1.exterior.is_ccw
+        assert polygon2.exterior.is_ccw
+        assert orient(multipolygon, 1) == MultiPolygon([polygon1_reversed, polygon2])
+        assert orient(multipolygon, -1) == MultiPolygon([polygon1, polygon2_reversed])
+
+    def test_geometrycollection(self):
+        polygon = Polygon([(0, 0), (0, 1), (1, 0)])
+        polygon_reversed = Polygon(polygon.exterior.coords[::-1])
+        collection = GeometryCollection([polygon])
+        assert orient(collection, 1) == GeometryCollection([polygon_reversed])
+        assert orient(collection, -1) == GeometryCollection([polygon])
+
+    def test_polygon_with_holes(self):
+        ring_cw = LinearRing([(0, 0), (0, 1), (1, 1), (0, 0)])
+        ring_cw2 = LinearRing([(0, 0), (0, 3), (3, 3), (0, 0)])
+        ring_ccw = LinearRing([(0, 0), (1, 1), (0, 1), (0, 0)])
+        ring_ccw2 = LinearRing([(0, 0), (2, 2), (0, 2), (0, 0)])
+
+        polygon_with_holes_mixed = Polygon(
+            ring_ccw, [ring_cw, ring_ccw2, ring_cw2, ring_ccw]
+        )
+        polygon_with_holes_ccw = Polygon(
+            ring_ccw, [ring_cw, ring_ccw2.reverse(), ring_cw2, ring_ccw.reverse()]
+        )
+
+        assert_array_equal(orient(polygon_with_holes_ccw, 1), polygon_with_holes_ccw)
+        assert_array_equal(
+            orient(polygon_with_holes_ccw, -1), polygon_with_holes_ccw.reverse()
+        )
+        assert_array_equal(orient(polygon_with_holes_mixed, 1), polygon_with_holes_ccw)
+        assert_array_equal(
+            orient(polygon_with_holes_mixed, -1), polygon_with_holes_ccw.reverse()
+        )