Changeset 4920

Timestamp:

Jan 9, 2008, 2:25:43 PM (17 years ago)

Author:

nick

Message:

added one more test for split in geospatial_data

File:

• anuga_core/source/anuga/geospatial_data/test_geospatial_data.py (modified) (2 diffs)

anuga_core/source/anuga/geospatial_data/test_geospatial_data.py

@@ -1820 +1820 @@
         A = G1.get_attributes()
         assert allclose(A,[24, 18, 17, 11, 8])
+        
+    def test_split1(self):
+        """test if the results from spilt are disjoin sets"""
+        from RandomArray import randint,seed
+        seed(100,100)
+        a_points = randint(0,999999,(10,2))
+        points = a_points.tolist()
+#        print points
+
+        G = Geospatial_data(points)
+
+        factor = 0.1
+
+        #will return G1 with 10% of points and G2 with 90%
+        G1, G2  = G.split(factor,100) 
+        
+#        print 'G1',G1
+        assert allclose(len(G), len(G1)+len(G2))
+        assert allclose(round(len(G)*factor), len(G1))
+
+        P = G1.get_data_points(absolute=False)
+        assert allclose(P, [[982420.,28233.]])
+
  
     def test_find_optimal_smoothing_parameter(self):
@@ -1939 +1962 @@
     #suite = unittest.makeSuite(Test_Geospatial_data, 'test_write_csv_attributes_lat_long')
 #    suite = unittest.makeSuite(Test_Geospatial_data, 'test_find_optimal_smoothing_parameter')
-#    suite = unittest.makeSuite(Test_Geospatial_data, 'test_split')
+#    suite = unittest.makeSuite(Test_Geospatial_data, 'test_split1')
     suite = unittest.makeSuite(Test_Geospatial_data, 'test')
     runner = unittest.TextTestRunner() #verbosity=2)