Changeset 1121

Timestamp:

Mar 22, 2005, 3:57:26 PM (20 years ago)

Author:

prow

Message:

 

File:

• inundation/ga/storm_surge/pyvolution/test_data_manager.py (modified) (1 diff)

inundation/ga/storm_surge/pyvolution/test_data_manager.py

@@ -707 +707 @@
         os.remove('small.sww')
 
+    def test_ferret2sww3(self):
+        """
+        """
+        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
+
+        fid1 = NetCDFFile('small_ha.nc')
+        fid2 = NetCDFFile('small_e.nc')
+        fid3 = NetCDFFile('small_va.nc')
+
+        first_amp = fid1.variables['HA'][:][0,0,0]
+        fourth_amp = fid1.variables['HA'][:][0,0,3]
+        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,3]
+
+        fid1.close()
+        fid2.close()
+        fid3.close()
+
+        #Call conversion (with zero origin)
+        ferret2sww('small', verbose=False,
+                   origin = (56, 0, 0))
+
+
+        #Work out the UTM coordinates for first point
+        zone, e, n = redfearn(-34.5, 150.66667)
+        #print zone, e, n
+
+        #Read output file 'small.sww'
+        fid = NetCDFFile('small.sww')
+
+        x = fid.variables['x'][:]
+        y = fid.variables['y'][:]
+        
+
+        #Check that first coordinate is correctly represented
+        assert allclose(x[0], e)
+        assert allclose(y[0], n)
+
+        #Check first value
+        stage = fid.variables['stage'][:]
+        xmomentum = fid.variables['xmomentum'][:]
+        ymomentum = fid.variables['ymomentum'][:]       
+
+        #print ymomentum
+        first_height = first_amp/100 - first_elevation
+        fourth_height = fourth_amp/100 - fourth_elevation
+
+        first_momentum=first_speed*first_height/100
+        fourth_momentum=fourth_speed*fourth_height/100
+
+        assert allclose(ymomentum[0][0],first_momentum)  #Meters
+        assert allclose(ymomentum[0][1],fourth_momentum)  #Meters
+
+        fid.close()
+
+        #Cleanup
+        import os
+        os.remove('small.sww')
+