Changeset 3983 for anuga_core/source/anuga/abstract_2d_finite_volumes/test_util.py

Timestamp:

Nov 14, 2006, 2:20:29 PM (17 years ago)

Author:

duncan

Message:

new function. Create a directory structure. Handy for writing data to.

File:

• anuga_core/source/anuga/abstract_2d_finite_volumes/test_util.py (modified) (2 diffs)

anuga_core/source/anuga/abstract_2d_finite_volumes/test_util.py

@@ -5 +5 @@
 from Numeric import zeros, array, allclose, Float
 from math import sqrt, pi
-import tempfile
+import tempfile, os
+from os import access, F_OK,sep, removedirs
 
 from anuga.abstract_2d_finite_volumes.util import *
@@ -1154 +1155 @@
         assert info.startswith('Revision')
                           
-                          
+        
+    def test_add_directories(self):
+        
+        import tempfile
+        root_dir = tempfile.mkdtemp('_test_util', 'test_util_')
+        directories = ['ja','ne','ke']
+        kens_dir = add_directories(root_dir, directories)
+        assert kens_dir == root_dir + sep + 'ja' + sep + 'ne' + \
+               sep + 'ke'
+        assert access(root_dir,F_OK)
+
+        #clean up!
+        os.rmdir(kens_dir)
+        os.rmdir(root_dir + sep + 'ja' + sep + 'ne')
+        os.rmdir(root_dir + sep + 'ja')
+        os.rmdir(root_dir)
 
 #-------------------------------------------------------------
 if __name__ == "__main__":
     suite = unittest.makeSuite(Test_Util,'test')
-    #suite = unittest.makeSuite(Test_Util,'test_get_version_info')
+    #suite = unittest.makeSuite(Test_Util,'test_add_directories')
     runner = unittest.TextTestRunner()
     runner.run(suite)