Changeset 4961

Timestamp:

Jan 21, 2008, 10:24:47 AM (17 years ago)

Author:

ole

Message:

Rewrote CRC test using NamedTemporaryFile?

File:

• anuga_core/source/anuga/utilities/test_system_tools.py (modified) (3 diffs)

anuga_core/source/anuga/utilities/test_system_tools.py

@@ -34 +34 @@
 
         import zlib
-        from tempfile import NamedTemporaryFile, mkstemp, mktemp
+        from tempfile import NamedTemporaryFile, mktemp
 
         # Generate a text file
-        fid = open(mktemp(suffix='.tmp',
-                          dir='.'), 'w')
+        fid = NamedTemporaryFile(mode='w',
+                                 suffix='.tmp',
+                                 dir='.')
         string = 'My temp file with textual content. AAAABBBBCCCC1234'
         fid.write(string)
-        fid.close()
+        fid.flush()
 
 
@@ -49 +50 @@
         assert checksum == ref_crc
 
-        os.remove(fid.name)
+        # Close and remove temporary file
+        fid.close()
+
+
 
         # Binary file
-        fid = open(mktemp(suffix='.tmp',
-                          dir='.'), 'wb')
-        string = 'My temp file with textual content. AAAABBBBCCCC1234'
+        fid = NamedTemporaryFile(mode='w+b',
+                                 suffix='.tmp',
+                                 dir='.')
+        string = 'My temp file with binary content. AAAABBBBCCCC1234'
         fid.write(string)
-        fid.close()
+        fid.flush()
+
 
         ref_crc = zlib.crc32(string)
@@ -62 +68 @@
 
         assert checksum == ref_crc
-        os.remove(fid.name)
+
+        # Close and remove temporary file
+        fid.close()
         
-        # Binary NetCDF File X 2
+        
+        # Binary NetCDF File X 2 (use mktemp's name)
 
         test_array = array([[7.0, 3.14], [-31.333, 0.0]])