Changeset 1854 for inundation/pyvolution/test_util.py

Timestamp:

Sep 28, 2005, 10:29:52 AM (19 years ago)

Author:

ole

Message:

Added functionality for excluding polygons in populate_polygon

File:

• inundation/pyvolution/test_util.py (modified) (1 diff)

inundation/pyvolution/test_util.py

@@ -1112 +1112 @@
 
 
+    def test_populate_polygon_with_exclude(self):
+        
+
+        polygon = [[0,0], [1,0], [1,1], [0,1]]
+        ex_poly = [[0,0], [0.5,0], [0.5, 0.5], [0,0.5]] #SW quarter
+        points = populate_polygon(polygon, 5, exclude = [ex_poly])
+
+        assert len(points) == 5
+        for point in points:
+            assert inside_polygon(point, polygon)
+            assert not inside_polygon(point, ex_poly)            
+
+
+        #overlap
+        polygon = [[0,0], [1,0], [1,1], [0,1]]
+        ex_poly = [[-1,-1], [0.5,0], [0.5, 0.5], [-1,0.5]]
+        points = populate_polygon(polygon, 5, exclude = [ex_poly])
+
+        assert len(points) == 5
+        for point in points:
+            assert inside_polygon(point, polygon)
+            assert not inside_polygon(point, ex_poly)                        
+        
+        #Multiple
+        polygon = [[0,0], [1,0], [1,1], [0,1]]
+        ex_poly1 = [[0,0], [0.5,0], [0.5, 0.5], [0,0.5]] #SW quarter
+        ex_poly2 = [[0.5,0.5], [0.5,1], [1, 1], [1,0.5]] #NE quarter        
+        
+        points = populate_polygon(polygon, 20, exclude = [ex_poly1, ex_poly2])
+
+        assert len(points) == 20
+        for point in points:
+            assert inside_polygon(point, polygon)
+            assert not inside_polygon(point, ex_poly1)
+            assert not inside_polygon(point, ex_poly2)                                
+        
+
+        #Very convoluted polygon
+        polygon = [[0,0], [10,10], [15,5], [20, 10], [25,0], [30,10], [40,-10]]
+        ex_poly = [[-1,-1], [5,0], [5, 5], [-1,5]]
+        points = populate_polygon(polygon, 20, exclude = [ex_poly])
+        
+        assert len(points) == 20
+        for point in points:
+            assert inside_polygon(point, polygon)
+            assert not inside_polygon(point, ex_poly), '%s' %str(point)                        
+
+
 
 #-------------------------------------------------------------