Changeset 2655 for inundation/fit_interpolate/test_interpolate.py

Timestamp:

Apr 4, 2006, 3:49:42 PM (18 years ago)

Author:

duncan

Message:

Adding ability to remove points outside of the mesh when interpolating

File:

• inundation/fit_interpolate/test_interpolate.py (modified) (12 diffs)

inundation/fit_interpolate/test_interpolate.py

@@ -20 +20 @@
 from coordinate_transforms.geo_reference import Geo_reference
 from shallow_water import Domain, Transmissive_boundary
-from utilities.numerical_tools import mean
+from utilities.numerical_tools import mean, INF
 from data_manager import get_dataobject
 from geospatial_data.geospatial_data import Geospatial_data
@@ -237 +237 @@
         assert allclose(sum(results, axis=1), 1.0)
 
-        #FIXME - have to change this test to check default info
-    def NO_test_arbitrary_datapoints_some_outside(self):
+    def test_arbitrary_datapoints_some_outside(self):
         """Try arbitrary datapoints one outside the triangle.
         That one should be ignored
@@ -253 +252 @@
         data = [ [0.2, 1.5], [0.123, 1.768], [1.43, 0.44], [5.0, 7.0]]
 
-
-        interp = Interpolate(points, vertices, data, precrop = True)
-        
-        results = interp._build_interpolation_matrix_A(data).todense()
-        assert allclose(sum(results, axis=1), 1.0)
-
-        interp = Interpolate(points, vertices, data, precrop = False)
+        interp = Interpolate(points, vertices, data)
         results = interp._build_interpolation_matrix_A(data).todense()
         assert allclose(sum(results, axis=1), [1,1,1,0])
@@ -301 +294 @@
         assert allclose(A, answer)
 
-
+        #FIXME -  tests are hopefully failing due to points outsdie the
+        # mesh
     def test_interpolate_attributes_to_points(self):
         v0 = [0.0, 0.0]
@@ -442 +436 @@
 
         z = interp.interpolate(f, point_coords)
-        answer = [ [0., 0., 0., 0.]] # (-1,-1)
+        answer = array([ [INF, INF, INF, INF]]) # (-1,-1)
 
         #print "***********"
@@ -452 +446 @@
         # of the mesh? Not currently.  
 
-        assert allclose(z, answer)
-
+        for i in range(4):
+            self.failUnless( z[0,i] == answer[0,i], 'Fail!')
+        
     def test_interpolate_attributes_to_pointsIV(self):
         a = [-1.0, 0.0]
@@ -594 +589 @@
 
     def test_interpolation_interface_time_only(self):
-        """Test spatio-temporal interpolation
-        Test that spatio temporal function performs the correct
-        interpolations in both time and space
-        """
+
+        # Test spatio-temporal interpolation
+        # Test that spatio temporal function performs the correct
+        # interpolations in both time and space
+        
 
 
@@ -667 +663 @@
 
     def test_interpolation_interface_spatial_only(self):
-        """Test spatio-temporal interpolation with constant time
-        """
-
+        # Test spatio-temporal interpolation with constant time
+        
         #Three timesteps
         time = [1.0, 5.0, 6.0]
@@ -728 +723 @@
 
     def test_interpolation_interface(self):
-        """Test spatio-temporal interpolation
-        Test that spatio temporal function performs the correct
-        interpolations in both time and space
-        """
-
+        # Test spatio-temporal interpolation
+        # Test that spatio temporal function performs the correct
+        # interpolations in both time and space
+    
 
         #Three timesteps
@@ -793 +787 @@
 
 
-    def qtest_interpolation_interface(self):
+    def test_points_outside_the_polygon(self):
         a = [-1.0, 0.0]
         b = [3.0, 4.0]
@@ -815 +809 @@
                         [-0.9, -1.5],
                         [0.5, -1.9],
-                        [999999, 9999999]]
+                        [999999, 9999999]] # point Outside poly
         geo_data = Geospatial_data(data_points = point_coords)
 
@@ -827 +821 @@
                   2*linear_function(point_coords) ]
         answer = transpose(answer)
+        answer[11,:] = [INF, INF]
         #print "z",z
-        #print "answer",answer
-        assert allclose(z, answer)
-
+        #print "answer _ fixed",answer
+        assert allclose(z[0:10], answer[0:10])
+        self.failUnless( z[11,1] == answer[11,1], 'Fail!')
+        self.failUnless( z[11,0] == answer[11,0], 'Fail!')
+
+        
+    def test_points_outside_the_polygon(self):
+        a = [-1.0, 0.0]
+        b = [3.0, 4.0]
+        c = [4.0, 1.0]
+        d = [-3.0, 2.0] #3
+        e = [-1.0, -2.0]
+        f = [1.0, -2.0] #5
+
+        vertices = [a, b, c, d,e,f]
+        triangles = [[0,1,3], [1,0,2], [0,4,5], [0,5,2]] #abd bac aef afc
+
+        point_coords = [[-2.0, 2.0],
+                        [-1.0, 1.0],
+                        [9999.0, 9999.0], # point Outside poly
+                        [-9999.0, 1.0], # point Outside poly
+                        [2.0, 1.0],
+                        [0.0, 0.0],
+                        [1.0, 0.0],
+                        [0.0, -1.0],
+                        [-0.2, -0.5],
+                        [-0.9, -1.5],
+                        [0.5, -1.9],
+                        [999999, 9999999]] # point Outside poly
+        geo_data = Geospatial_data(data_points = point_coords)
+
+        interp = Interpolate(vertices, triangles)
+        f = array([linear_function(vertices),2*linear_function(vertices) ])
+        f = transpose(f)
+        #print "f",f
+        z = interp.interpolate(f, geo_data)
+        #z = interp.interpolate(f, point_coords)
+        answer = [linear_function(point_coords),
+                  2*linear_function(point_coords) ]
+        answer = transpose(answer)
+        answer[2,:] = [INF, INF]
+        answer[3,:] = [INF, INF]
+        answer[11,:] = [INF, INF]
+        #print "z",z
+        #print "answer _ fixed",answer
+        assert allclose(z[0:1], answer[0:1])
+        assert allclose(z[4:10], answer[4:10])
+        for i in [2,3,11]:
+            self.failUnless( z[i,1] == answer[11,1], 'Fail!')
+            self.failUnless( z[i,0] == answer[11,0], 'Fail!')
         
 #-------------------------------------------------------------
 if __name__ == "__main__":
+
+    #suite = unittest.makeSuite(Test_Interpolate,'test')
     suite = unittest.makeSuite(Test_Interpolate,'test')
+    #suite = unittest.makeSuite(Test_Interpolate,'test_points_outside_the_polygon')
     runner = unittest.TextTestRunner(verbosity=1)
     runner.run(suite)