Changeset 4956

Timestamp:

Jan 18, 2008, 3:58:24 PM (16 years ago)

Author:

ole

Message:

Work on CRC32 for IP tracking

File:

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

anuga_core/source/anuga/utilities/test_system_tools.py

@@ -3 +3 @@
 
 import unittest
-from Numeric import zeros, array, allclose
+from Numeric import zeros, array, allclose, Float
+from Scientific.IO.NetCDF import NetCDFFile
 
 from system_tools import *
@@ -25 +26 @@
         # print host
         assert isinstance(host, basestring), 'User name should be a string'        
+
     def test_compute_checksum(self):
         """test_compute_checksum(self):
@@ -30 +32 @@
         Check that checksums on files are OK
         """
-        # FIXME: Not Done Yet
 
-        from tempfile import mkstemp
+        import zlib
+        from tempfile import NamedTemporaryFile, mkstemp, mktemp
 
         # Generate a text file
-        fd, pathname = mkstemp('.tmp', '', '.', '')
+        fid = open(mktemp(suffix='.tmp',
+                          dir='.'), 'w')
+        string = 'My temp file with textual content. AAAABBBBCCCC1234'
+        fid.write(string)
+        fid.close()
+
+
+        ref_crc = zlib.crc32(string)
+        checksum = compute_checksum(fid.name)
+
+        assert checksum == ref_crc
+
+        os.remove(fid.name)
+
+        # Binary file
+        fid = open(mktemp(suffix='.tmp',
+                          dir='.'), 'wb')
+        string = 'My temp file with textual content. AAAABBBBCCCC1234'
+        fid.write(string)
+        fid.close()
+
+        ref_crc = zlib.crc32(string)
+        checksum = compute_checksum(fid.name)
+
+        assert checksum == ref_crc
+        os.remove(fid.name)
+        
+        # Binary NetCDF File X 2
+
+        test_array = array([[7.0, 3.14], [-31.333, 0.0]])
+
+        # First file
+        filename1 = mktemp(suffix='.nc', dir='.')
+        fid = NetCDFFile(filename1, 'w')
+        fid.createDimension('two', 2)
+        fid.createVariable('test_array', Float,
+                           ('two', 'two'))
+        fid.variables['test_array'][:] = test_array
+        fid.close()
+
+        # Second file
+        filename2 = mktemp(suffix='.nc', dir='.')
+        fid = NetCDFFile(filename2, 'w')
+        fid.createDimension('two', 2)
+        fid.createVariable('test_array', Float,
+                           ('two', 'two'))
+        fid.variables['test_array'][:] = test_array
+        fid.close()
 
         
-        
+        checksum1 = compute_checksum(filename1)
+        checksum2 = compute_checksum(filename2)        
+        assert checksum1 == checksum2
+
+
+        os.remove(filename1)
+        os.remove(filename2)