Changeset 1394 for inundation/ga/storm_surge/pmesh/load_mesh/test_loadASCII.py

Timestamp:

May 16, 2005, 11:06:22 AM (19 years ago)

Author:

duncan

Message:

refactoring dealing with point files

File:

• inundation/ga/storm_surge/pmesh/load_mesh/test_loadASCII.py (modified) (4 diffs)

inundation/ga/storm_surge/pmesh/load_mesh/test_loadASCII.py

@@ -163 +163 @@
         pass
 
-    
-    def FUNCTION_REMOVEDtest_import_mesh(self):
-        
-        dict = self.dict
-        fileName = tempfile.mktemp(".txt")
-        fd = open(fileName,'w')
-        loadASCII._write_ASCII_outline(fd,dict)
-        fd.close()
-        loaded_dict = import_mesh(fileName) #FIXME function names are wacky
-        os.remove(fileName)
-        
-        #gen_dict = 
-        #print "*********************"
-        #print dict
-        #print "**loaded_dict*******************"
-        #print loaded_dict
-        #print "*********************"
-
-        
-        self.failUnless(array(loaded_dict['points'])  ==
-                        array(dict['points']),
-                         'test_import_mesh failed. Test 1')
-        self.failUnless(array(loaded_dict['point_attributes'])  ==
-                        array(dict['point_attributes']),
-                         'test_import_mesh failed. Test 2')
-        self.failUnless(array(loaded_dict['outline_segments'])  ==
-                        array(dict['outline_segments']),
-                         'test_import_mesh failed. Test 3')
-        self.failUnless(array(loaded_dict['outline_segment_tags'])  ==
-                        array(dict['outline_segment_tags']),
-                         'test_import_mesh failed. Test 4')
-        self.failUnless(array(loaded_dict['regions'])  ==
-                        array(dict['regions']),
-                         'test_import_mesh failed. Test 5')
-        self.failUnless(array(loaded_dict['region_tags'])  ==
-                        array(dict['region_tags']),
-                         'test_import_mesh failed. Test 5')
-        self.failUnless(array(loaded_dict['region_max_areas'])  ==
-                        array(dict['region_max_areas']),
-                         'test_import_mesh failed. Test 5')
-
-        self.failUnless(array(loaded_dict['holes'])  ==
-                        array(dict['holes']),
-                         'test_import_mesh failed. Test 6')
         
     def test_export_mesh_file(self):
@@ -486 +442 @@
                          'test_writepts failed. Test geo_reference')
 
+    def test_write_xya(self):
+        dict = {}
+        att_dict = {}
+        dict['pointlist'] = array([[1.0, 0.0],[0.0, 1.0],[1.0, 0.0]])
+        att_dict['elevation'] = array([10.0, 0.0, 10.4])
+        att_dict['brightness'] = array([10.0, 0.0, 10.4])
+        dict['attributelist'] = att_dict
+        dict['geo_reference'] = Geo_reference(56,1.9,1.9)
+        
+        
+        fileName = tempfile.mktemp(".xya")
+        export_points_file(fileName, dict)
+        dict2 = import_points_file(fileName)
+        #print "fileName",fileName 
+        os.remove(fileName)
+        #print "dict2",dict2 
+        
+        assert allclose(dict2['pointlist'],[[1.0, 0.0],[0.0, 1.0],[1.0, 0.0]])
+        assert allclose(dict2['attributelist']['elevation'], [10.0, 0.0, 10.4])
+        answer = [10.0, 0.0, 10.4]
+        assert allclose(dict2['attributelist']['brightness'], answer)
+
         
     def test_half_pts(self):
@@ -615 +593 @@
         try:
             dict = import_points_file(fileName,delimiter = ' ')
-        except TitleAmountError:
+        except IOError:
             pass
         else:
@@ -622 +600 @@
         
         os.remove(fileName)
+       
+    def test_loadxy6(self):
+        import os
+        import tempfile
+       
+        fileName = tempfile.mktemp(".xya")
+        file = open(fileName,"w")
+        file.write("elevation\n\
+1.0 0.0 10.0 \n\
+0.0 1.0\n\
+1.0 \n")
+        file.close()
+        #print fileName
+        try:
+            dict = import_points_file(fileName,delimiter = ' ')
+        except IOError:
+            pass
+        else:
+            self.failUnless(0 ==1,
+                        'bad xya file did not raise error!')
+        
+        os.remove(fileName)
          
+ 
+    def test_import_tsh(self):
+        import os
+        import tempfile
+       
+        fileName = tempfile.mktemp(".tsh")
+        file = open(fileName,"w")
+        file.write("elevation\n\
+1.0 0.0 10.0 \n\
+0.0 1.0\n\
+1.0 \n")
+        file.close()
+        #print fileName
+        try:
+            dict = import_points_file(fileName,delimiter = ' ')
+        except IOError:
+            pass
+        else:
+            self.failUnless(0 ==1,
+                        'bad tsh file did not raise error!')
+        
+        os.remove(fileName)
+  
+    def test_import_tsh2(self):
+        import os
+        import tempfile
+       
+        fileName = tempfile.mktemp(".msh")
+        file = open(fileName,"w")
+        file.write("1.0 \n\
+1.0 0.0 10.0 \n\
+0.0 1.0\n\
+13.0 \n")
+        file.close()
+        #print fileName
+        try:
+            dict = import_points_file(fileName,delimiter = ' ')
+        except IOError:
+            pass
+        else:
+            self.failUnless(0 ==1,
+                        'bad tsh file did not raise error!')
+        
+        os.remove(fileName)         
+ 
+    def test_import_tsh3(self):
+        import os
+        import tempfile
+       
+        fileName = tempfile.mktemp(".tsh")
+        file = open(fileName,"w")
+        file.write("1.0 \n\
+showme1.0 0.0 10.0 \n\
+0.0 1.0\n\
+13.0 \n")
+        file.close()
+        #print fileName
+        try:
+            dict = import_points_file(fileName,delimiter = ' ')
+        except IOError:
+            pass
+        else:
+            self.failUnless(0 ==1,
+                        'bad tsh file did not raise error!')
+        
+        os.remove(fileName)         
  
 #-------------------------------------------------------------