Changeset 1132

Timestamp:

Mar 23, 2005, 10:00:15 AM (20 years ago)

Author:

prow

Message:

non-trivial test_ferret2sww

File:

• inundation/ga/storm_surge/pyvolution/test_data_manager.py (modified) (5 diffs)

inundation/ga/storm_surge/pyvolution/test_data_manager.py

@@ -707 +707 @@
         os.remove('small.sww')
 
+
+
     def test_ferret2sww3(self):
         """
@@ -725 +727 @@
         # Fourth value (index==3) is -6.50198 cm
 
-
         from coordinate_transforms.redfearn import redfearn
 
-        fid1 = NetCDFFile('small_ha.nc','a')
-        fid3 = NetCDFFile('small_va.nc','a')
-
-        #Populate with decent values
-        fid1.variables['HA'][:][0,0,3]=130.7
-        fid3.variables['VA'][:][0,0,3]=10.0
-
-
+        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 = 'LON'
+        lat_name = 'LAT'
+
+        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',2)
+            fid.createVariable('TIME','d',('TIME',))
+            fid.variables['TIME'].point_spacing='uneven'
+            fid.variables['TIME'].units='seconds'
+            fid.variables['TIME'].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',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()
-
-        fid1 = NetCDFFile('small_ha.nc','r')
-        fid2 = NetCDFFile('small_e.nc','r')
-        fid3 = NetCDFFile('small_va.nc','r')
+        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]
-        fourth_amp = fid1.variables['HA'][:][0,0,3]
+        third_amp = fid1.variables['HA'][:][0,0,2]
         first_elevation = fid2.variables['ELEVATION'][0,0]
-        fourth_elevation= fid2.variables['ELEVATION'][:][0,3]
-        first_speed = fid3.variables['VA'][0,0]
-        fourth_speed = fid3.variables['VA'][:][0,0,3]
-        #print 'fourth speed', fourth_speed
+        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()
@@ -756 +833 @@
 
         #Call conversion (with zero origin)
-        ferret2sww('small', verbose=False,
+        ferret2sww('test', verbose=False,
                    origin = (56, 0, 0))
 
 
-        #Read output file 'small.sww'
-        fid = NetCDFFile('small.sww')
+        #Read output file 'test.sww'
+        fid = NetCDFFile('test.sww')
 
 
         #Check first value
+        elevation = fid.variables['elevation'][:]
         stage = fid.variables['stage'][:]
         xmomentum = fid.variables['xmomentum'][:]
@@ -771 +849 @@
         #print ymomentum
         first_height = first_amp/100 - first_elevation
-        fourth_height = fourth_amp/100 - fourth_elevation
-
+        third_height = third_amp/100 - third_elevation
         first_momentum=first_speed*first_height/100
-        fourth_momentum=fourth_speed*fourth_height/100
+        third_momentum=third_speed*third_height/100
 
         assert allclose(ymomentum[0][0],first_momentum)  #Meters
-
-        #print 'YM1', ymomentum[0][1]
-        #print 'YM4', ymomentum[0][3]        
-        #print '4M', fourth_momentum
-        assert allclose(ymomentum[0][1],fourth_momentum)  #Meters
+        assert allclose(ymomentum[0][2],third_momentum)  #Meters
 
         fid.close()
@@ -787 +860 @@
         #Cleanup
         import os
-        os.remove('small.sww')
+        os.remove('test.sww')
+        os.remove('test_va.nc')
+        os.remove('test_ua.nc')
+        os.remove('test_ha.nc')