Changeset 2639

Timestamp:

Mar 31, 2006, 12:38:27 PM (17 years ago)

Author:

ole

Message:

Added new test for ferret generated NetCDF. This test breaks deliberately, since the code has yet to be written.

File:

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

inundation/pyvolution/test_data_manager.py

@@ -2213 +2213 @@
         long_name = 'LON'
         lat_name = 'LAT'
+        time_name = 'TIME'        
 
         nx = 3
@@ -2230 +2231 @@
             fid.variables[lat_name].assignValue(h2_list)
 
-            fid.createDimension('TIME',2)
-            fid.createVariable('TIME','d',('TIME',))
-            fid.variables['TIME'].point_spacing='uneven'
-            fid.variables['TIME'].units='seconds'
-            fid.variables['TIME'].assignValue([0.,1.])
+            fid.createDimension(time_name,2)
+            fid.createVariable(time_name,'d',(time_name,))
+            fid.variables[time_name].point_spacing='uneven'
+            fid.variables[time_name].units='seconds'
+            fid.variables[time_name].assignValue([0.,1.])
             if fid == fid3: break
 
@@ -2270 +2271 @@
 
         for fid in [fid1,fid2,fid3]:
-          fid.createVariable(name[fid],'d',('TIME',lat_name,long_name))
+          fid.createVariable(name[fid],'d',(time_name,lat_name,long_name))
+          fid.variables[name[fid]].point_spacing='uneven'
+          fid.variables[name[fid]].units=units[fid]
+          fid.variables[name[fid]].assignValue(values[fid])
+          fid.variables[name[fid]].missing_value = -99999999.
+          if fid == fid3: break
+
+        for fid in [fid4]:
+            fid.createVariable(name[fid],'d',(lat_name,long_name))
+            fid.variables[name[fid]].point_spacing='uneven'
+            fid.variables[name[fid]].units=units[fid]
+            fid.variables[name[fid]].assignValue(values[fid])
+            fid.variables[name[fid]].missing_value = -99999999.
+
+
+        fid1.sync(); fid1.close()
+        fid2.sync(); fid2.close()
+        fid3.sync(); fid3.close()
+        fid4.sync(); fid4.close()
+
+        fid1 = NetCDFFile('test_ha.nc','r')
+        fid2 = NetCDFFile('test_e.nc','r')
+        fid3 = NetCDFFile('test_va.nc','r')
+
+
+        first_amp = fid1.variables['HA'][:][0,0,0]
+        third_amp = fid1.variables['HA'][:][0,0,2]
+        first_elevation = fid2.variables['ELEVATION'][0,0]
+        third_elevation= fid2.variables['ELEVATION'][:][0,2]
+        first_speed = fid3.variables['VA'][0,0,0]
+        third_speed = fid3.variables['VA'][:][0,0,2]
+
+        fid1.close()
+        fid2.close()
+        fid3.close()
+
+        #Call conversion (with zero origin)
+        ferret2sww('test', verbose=False,
+                   origin = (56, 0, 0))
+
+        os.remove('test_va.nc')
+        os.remove('test_ua.nc')
+        os.remove('test_ha.nc')
+        os.remove('test_e.nc')
+
+        #Read output file 'test.sww'
+        fid = NetCDFFile('test.sww')
+
+
+        #Check first value
+        elevation = fid.variables['elevation'][:]
+        stage = fid.variables['stage'][:]
+        xmomentum = fid.variables['xmomentum'][:]
+        ymomentum = fid.variables['ymomentum'][:]
+
+        #print ymomentum
+        first_height = first_amp/100 - first_elevation
+        third_height = third_amp/100 - third_elevation
+        first_momentum=first_speed*first_height/100
+        third_momentum=third_speed*third_height/100
+
+        assert allclose(ymomentum[0][0],first_momentum)  #Meters
+        assert allclose(ymomentum[0][2],third_momentum)  #Meters
+
+        fid.close()
+
+        #Cleanup
+        os.remove('test.sww')
+
+
+
+    def test_ferret2sww4(self):
+        """Like previous but with augmented variable names as
+        in files produced by ferret as opposed to MOST
+        """
+        from Scientific.IO.NetCDF import NetCDFFile
+
+        #The test file has
+        # LON = 150.66667, 150.83334, 151, 151.16667
+        # LAT = -34.5, -34.33333, -34.16667, -34 ;
+        # ELEVATION = [-1 -2 -3 -4
+        #             -5 -6 -7 -8
+        #              ...
+        #              ...      -16]
+        # where the top left corner is -1m,
+        # and the ll corner is -13.0m
+        #
+        # First value (index=0) in small_ha.nc is 0.3400644 cm,
+        # Fourth value (index==3) is -6.50198 cm
+
+        from coordinate_transforms.redfearn import redfearn
+        import os
+        fid1 = NetCDFFile('test_ha.nc','w')
+        fid2 = NetCDFFile('test_ua.nc','w')
+        fid3 = NetCDFFile('test_va.nc','w')
+        fid4 = NetCDFFile('test_e.nc','w')
+
+        h1_list = [150.66667,150.83334,151.]
+        h2_list = [-34.5,-34.33333]
+
+        long_name = 'LON961_1261'
+        lat_name = 'LAT481_841'
+        time_name = 'TIME1'
+
+        nx = 3
+        ny = 2
+
+        for fid in [fid1,fid2,fid3]:
+            fid.createDimension(long_name,nx)
+            fid.createVariable(long_name,'d',(long_name,))
+            fid.variables[long_name].point_spacing='uneven'
+            fid.variables[long_name].units='degrees_east'
+            fid.variables[long_name].assignValue(h1_list)
+
+            fid.createDimension(lat_name,ny)
+            fid.createVariable(lat_name,'d',(lat_name,))
+            fid.variables[lat_name].point_spacing='uneven'
+            fid.variables[lat_name].units='degrees_north'
+            fid.variables[lat_name].assignValue(h2_list)
+
+            fid.createDimension(time_name,2)
+            fid.createVariable(time_name,'d',(time_name,))
+            fid.variables[time_name].point_spacing='uneven'
+            fid.variables[time_name].units='seconds'
+            fid.variables[time_name].assignValue([0.,1.])
+            if fid == fid3: break
+
+
+        for fid in [fid4]:
+            fid.createDimension(long_name,nx)
+            fid.createVariable(long_name,'d',(long_name,))
+            fid.variables[long_name].point_spacing='uneven'
+            fid.variables[long_name].units='degrees_east'
+            fid.variables[long_name].assignValue(h1_list)
+
+            fid.createDimension(lat_name,ny)
+            fid.createVariable(lat_name,'d',(lat_name,))
+            fid.variables[lat_name].point_spacing='uneven'
+            fid.variables[lat_name].units='degrees_north'
+            fid.variables[lat_name].assignValue(h2_list)
+
+        name = {}
+        name[fid1]='HA'
+        name[fid2]='UA'
+        name[fid3]='VA'
+        name[fid4]='ELEVATION'
+
+        units = {}
+        units[fid1]='cm'
+        units[fid2]='cm/s'
+        units[fid3]='cm/s'
+        units[fid4]='m'
+
+        values = {}
+        values[fid1]=[[[5., 10.,15.], [13.,18.,23.]],[[50.,100.,150.],[130.,180.,230.]]]
+        values[fid2]=[[[1., 2.,3.], [4.,5.,6.]],[[7.,8.,9.],[10.,11.,12.]]]
+        values[fid3]=[[[13., 12.,11.], [10.,9.,8.]],[[7.,6.,5.],[4.,3.,2.]]]
+        values[fid4]=[[-3000,-3100,-3200],[-4000,-5000,-6000]]
+
+        for fid in [fid1,fid2,fid3]:
+          fid.createVariable(name[fid],'d',(time_name,lat_name,long_name))
           fid.variables[name[fid]].point_spacing='uneven'
           fid.variables[name[fid]].units=units[fid]