Changeset 6428 for branches/numpy/anuga/coordinate_transforms/test_geo_reference.py

Timestamp:

Feb 27, 2009, 11:54:09 AM (16 years ago)

Author:

rwilson

Message:

numpy changes

File:

• branches/numpy/anuga/coordinate_transforms/test_geo_reference.py (modified) (1 diff)

branches/numpy/anuga/coordinate_transforms/test_geo_reference.py

@@ -503 +503 @@
                         'bad shape did not raise error!')
             os.remove(point_file)
+
+    def test_functionality_get_absolute(self):
+        x0 = 1000.0
+        y0 = 2000.0
+        geo = Geo_reference(56, x0, y0)
+
+        # iterable points (*not* num.array())
+        points = ((2,3), (3,1), (5,2))
+        abs_points = geo.get_absolute(points)
+        # check we haven't changed 'points' itself
+        self.failIf(num.alltrue(abs_points == points))
+        new_points = abs_points.copy()
+        new_points[:,0] -= x0
+        new_points[:,1] -= y0
+        self.failUnless(num.alltrue(new_points == points))
+
+        # points in num.array()
+        points = num.array(((2,3), (3,1), (5,2)), num.float)
+        abs_points = geo.get_absolute(points)
+        # check we haven't changed 'points' itself
+        self.failIf(num.alltrue(abs_points == points))
+        new_points = abs_points.copy()
+        new_points[:,0] -= x0
+        new_points[:,1] -= y0
+        self.failUnless(num.alltrue(new_points == points))
+
         
 #-------------------------------------------------------------
 
 if __name__ == "__main__":
-    suite = unittest.makeSuite(geo_referenceTestCase,'test')
+    suite = unittest.makeSuite(geo_referenceTestCase, 'test')
+    #suite = unittest.makeSuite(geo_referenceTestCase, 'test_functionality_get_absolute')
     runner = unittest.TextTestRunner() #verbosity=2)
     runner.run(suite)