Changeset 1802

Timestamp:

Sep 8, 2005, 10:28:05 AM (18 years ago)

Author:

tdhu

Message:

Updated to remove any files created as part of the testing process

File:

• inundation/pyvolution/test_ermapper.py (modified) (3 diffs)

inundation/pyvolution/test_ermapper.py

@@ -6 +6 @@
 import ermapper_grids
 import Numeric
+from os import remove
 
 class Test_ERMapper(unittest.TestCase):
@@ -28 +29 @@
         assert Numeric.allclose(original_grid, new_grid)
 
+        # Clean up created files
+        remove(filename)
+        
     def test_basic_single_line_grid_default_format(self):
         # Setup test data
@@ -42 +46 @@
         assert Numeric.allclose(original_grid, new_grid)
 
-
+        # Clean up created files
+        remove(filename)
+        
     def test_write_default_header(self):
-        filename = 'test_write_ermapper_grid'
+        data_filename = 'test_write_ermapper_grid'
 
         # setup test data
         original_grid = Numeric.array([[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]])
         # Write test data
-        ermapper_grids.write_ermapper_data(original_grid, filename)
+        ermapper_grids.write_ermapper_data(original_grid, data_filename)
         # Write test header
-        filename = filename + '.ers'                              
-        ermapper_grids.write_ermapper_header(filename)
+        header_filename = data_filename + '.ers'                              
+        ermapper_grids.write_ermapper_header(header_filename)
+
+        # Clean up created files
+        remove(data_filename)
+        remove(header_filename)
 
 # def test_default_filenames