Changeset 8032 for trunk/anuga_core/source/anuga/geometry

Timestamp:

Oct 10, 2010, 5:14:24 PM (14 years ago)

Author:

habili

Message:

Added tests for polygon_overlap and not_polygon_overlap

File:

• trunk/anuga_core/source/anuga/geometry/test_polygon.py (modified) (2 diffs)

trunk/anuga_core/source/anuga/geometry/test_polygon.py

@@ -12 +12 @@
                     read_polygon, point_on_line, point_in_polygon, \
                     plot_polygons, outside_polygon, is_outside_polygon, \
-                    intersection, is_complex, is_inside_triangle, \
-                    interpolate_polyline, inside_polygon, in_and_outside_polygon
+                    intersection, is_complex, polygon_overlap, not_polygon_overlap,\
+                    is_inside_triangle, interpolate_polyline, inside_polygon, \
+                    in_and_outside_polygon
                     
 from polygon_function import Polygon_function
@@ -415 +416 @@
 
 
+    def test_polygon_overlap(self):
+        #rectangular polygon, 2 triangles overlap
+        polygon = [[3.,3.],[3.,4.],[5.,4.],[5.,3.]]
+        triangles = [[7.,10.],#does not overlap
+                     [8.,12.],
+                     [9.,10.],
+                     [3., 2.],#intersect
+                     [4., 5.],
+                     [5., 3.],
+                     [3., 2.],#intersect
+                     [4., 5.],
+                     [5., 2.],
+                     [0., 1.],#polygon inside triangle
+                     [5., 50.],
+                     [10., 1.],
+                     [3.5, 3.25],#triangle inside polygon
+                     [4., 3.75],
+                     [4.5, 3.25]]
+                     
+        indices = polygon_overlap(triangles, polygon)        
+        assert num.allclose([1, 2, 3, 4], indices)
+                     
+        #5 sided polygon, 2 triangles overlap
+        polygon = [[3.,3.],[3.,4.],[5.,4.],[5.5, 3.5],[5.,3.]]
+        triangles = [[7.,10.],#does not overlap
+                     [8.,12.],
+                     [9.,10.],
+                     [3., 2.],#intersect
+                     [4., 5.],
+                     [5., 3.],
+                     [3., 2.],#intersect
+                     [4., 5.],
+                     [5., 2.],
+                     [0., 1.],#polygon inside triangle
+                     [5., 50.],
+                     [10., 1.],
+                     [3.5, 3.25],#triangle inside polygon
+                     [4., 3.75],
+                     [4.5, 3.25]]
+    
+        indices = polygon_overlap(triangles, polygon)
+        assert num.allclose([1, 2, 3, 4], indices)
+              
+    def test_not_polygon_overlap(self):
+        #rectangular polygon, 2 triangles overlap
+        polygon = [[3.,3.],[3.,4.],[5.,4.],[5.,3.]]
+        triangles = [[7.,10.],#does not overlap
+                     [8.,12.],
+                     [9.,10.],
+                     [3., 2.],#intersect
+                     [4., 5.],
+                     [5., 3.],
+                     [3., 2.],#intersect
+                     [4., 5.],
+                     [5., 2.],
+                     [0., 1.],#polygon inside triangle
+                     [5., 50.],
+                     [10., 1.],
+                     [3.5, 3.25],#triangle inside polygon
+                     [4., 3.75],
+                     [4.5, 3.25]]
+                     
+        indices = not_polygon_overlap(triangles, polygon)        
+        assert num.allclose([0], indices)
+                     
+        #5 sided polygon, 2 triangles overlap
+        polygon = [[3.,3.],[3.,4.],[5.,4.],[5.5, 3.5],[5.,3.]]
+        triangles = [[7.,10.],#does not overlap
+                     [8.,12.],
+                     [9.,10.],
+                     [3., 2.],#intersect
+                     [4., 5.],
+                     [5., 3.],
+                     [3., 2.],#intersect
+                     [4., 5.],
+                     [5., 2.],
+                     [0., 1.],#polygon inside triangle
+                     [5., 50.],
+                     [10., 1.],
+                     [3.5, 3.25],#triangle inside polygon
+                     [4., 3.75],
+                     [4.5, 3.25]]
+    
+        indices = not_polygon_overlap(triangles, polygon)
+        assert num.allclose([0], indices)             
+                     
     def test_is_inside_triangle(self):
         # Simplest case: