Changeset 5398


Ignore:
Timestamp:
Jun 10, 2008, 4:12:17 PM (14 years ago)
Author:
duncan
Message:

Generalising to work in EQRM

File:
1 edited

Legend:

Unmodified
Added
Removed
  • anuga_core/source/anuga/utilities/test_system_tools.py

    r5219 r5398  
    44import unittest
    55from Numeric import zeros, array, allclose, Float
    6 from Scientific.IO.NetCDF import NetCDFFile
    76import zlib
    87from os.path import join, split, sep
     
    7675        # Binary NetCDF File X 2 (use mktemp's name)
    7776
    78         test_array = array([[7.0, 3.14], [-31.333, 0.0]])
     77        try:
     78            from Scientific.IO.NetCDF import NetCDFFile
     79        except ImportError:
     80            # This code is also used by EQRM which does not require NetCDF
     81            pass
     82        else:
     83            test_array = array([[7.0, 3.14], [-31.333, 0.0]])
    7984
    80         # First file
    81         filename1 = mktemp(suffix='.nc', dir='.')
    82         fid = NetCDFFile(filename1, 'w')
    83         fid.createDimension('two', 2)
    84         fid.createVariable('test_array', Float,
    85                            ('two', 'two'))
    86         fid.variables['test_array'][:] = test_array
    87         fid.close()
    88 
    89         # Second file
    90         filename2 = mktemp(suffix='.nc', dir='.')
    91         fid = NetCDFFile(filename2, 'w')
    92         fid.createDimension('two', 2)
    93         fid.createVariable('test_array', Float,
    94                            ('two', 'two'))
    95         fid.variables['test_array'][:] = test_array
    96         fid.close()
    97 
    98        
    99         checksum1 = compute_checksum(filename1)
    100         checksum2 = compute_checksum(filename2)       
    101         assert checksum1 == checksum2
     85            # First file
     86            filename1 = mktemp(suffix='.nc', dir='.')
     87            fid = NetCDFFile(filename1, 'w')
     88            fid.createDimension('two', 2)
     89            fid.createVariable('test_array', Float,
     90                               ('two', 'two'))
     91            fid.variables['test_array'][:] = test_array
     92            fid.close()
     93           
     94            # Second file
     95            filename2 = mktemp(suffix='.nc', dir='.')
     96            fid = NetCDFFile(filename2, 'w')
     97            fid.createDimension('two', 2)
     98            fid.createVariable('test_array', Float,
     99                               ('two', 'two'))
     100            fid.variables['test_array'][:] = test_array
     101            fid.close()
     102           
     103           
     104            checksum1 = compute_checksum(filename1)
     105            checksum2 = compute_checksum(filename2)       
     106            assert checksum1 == checksum2
    102107
    103108
    104         os.remove(filename1)
    105         os.remove(filename2)
     109            os.remove(filename1)
     110            os.remove(filename2)
    106111
    107112
Note: See TracChangeset for help on using the changeset viewer.