Changeset 2273

Timestamp:

Jan 24, 2006, 5:40:50 PM (19 years ago)

Author:

ole

Message:

Updated generation of test files

File:

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

inundation/pyvolution/test_data_manager.py

@@ -72 +72 @@
 
 
-        #Write A testfile
-
-        self.test_MOST_file = 'xxxsmall_ha.nc'
+        #Write A testfile (not realistic. Values aren't realistic)
+
+        self.test_MOST_file = 'most_small'
 
         longitudes = [150.66667, 150.83334, 151., 151.16667]
@@ -87 +87 @@
 
 
-        fid = NetCDFFile(self.test_MOST_file, 'w')
-        
-        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(longitudes)
-
-        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(latitudes)
-
-        fid.createDimension('TIME',six)
-        fid.createVariable('TIME','d',('TIME',))
-        fid.variables['TIME'].point_spacing='uneven'
-        fid.variables['TIME'].units='seconds'
-        fid.variables['TIME'].assignValue([0.0, 0.1, 0.6, 1.1, 1.6, 2.1])
-
-        fid.createVariable('HA','d',('TIME', lat_name, long_name))
-        #fid.createVariable('HA','d',(lat_name, long_name))
-        fid.variables['HA'].assignValue([[[0.3400644, 0, -46.63519, -6.50198],
-                                          [-0.1214216, 0, 0, 0],
-                                          [0, 0, 0, 0],
-                                          [0, 0, 0, 0]],
-                                         [[0.3400644, 2.291054e-005, -23.33335, -6.50198],
-                                          [-0.1213987, 4.581959e-005, -1.594838e-007, 1.421085e-012],
-                                          [2.291054e-005, 4.582107e-005, 4.581715e-005, 1.854517e-009],
-                                          [0, 2.291054e-005, 2.291054e-005, 0]],
-                                         [[0.3400644, 0.0001374632, -23.31503, -6.50198],
-                                          [-0.1212842, 0.0002756907, 0.006325484, 1.380492e-006],
-                                          [0.0001374632, 0.0002749264, 0.0002742863, 6.665601e-008],
-                                          [0, 0.0001374632, 0.0001374632, 0]],
-                                         [[0.3400644, 0.0002520159, -23.29672, -6.50198],
-                                          [-0.1211696, 0.0005075303, 0.01264618, 6.208276e-006],
-                                          [0.0002520159, 0.0005040318, 0.0005027961, 2.23865e-007],
-                                          [0, 0.0002520159, 0.0002520159, 0]],
-                                         [[0.3400644, 0.0003665686, -23.27842, -6.50198],
-                                          [-0.1210551, 0.0007413362, 0.01896192, 1.447638e-005],
-                                          [0.0003665686, 0.0007331371, 0.0007313463, 4.734126e-007],
-                                          [0, 0.0003665686, 0.0003665686, 0]],
-                                         [[0.3400644, 0.0004811212, -23.26012, -6.50198],
-                                          [-0.1209405, 0.0009771062, 0.02527271, 2.617787e-005],
-                                          [0.0004811212, 0.0009622425, 0.0009599366, 8.152277e-007],
-                                          [0, 0.0004811212, 0.0004811212, 0]]])
+        for ext in ['_ha.nc', '_ua.nc', '_va.nc', '_e.nc']:
+            fid = NetCDFFile(self.test_MOST_file + ext, 'w')
+       
+            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(longitudes)
+
+            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(latitudes)
+
+            fid.createDimension('TIME',six)
+            fid.createVariable('TIME','d',('TIME',))
+            fid.variables['TIME'].point_spacing='uneven'
+            fid.variables['TIME'].units='seconds'
+            fid.variables['TIME'].assignValue([0.0, 0.1, 0.6, 1.1, 1.6, 2.1])
+            
+
+            name = ext[1:3].upper()
+            if name == 'E.': name = 'ELEVATION'
+            fid.createVariable(name,'d',('TIME', lat_name, long_name))
+            fid.variables[name].units='CENTIMETERS'
+            fid.variables[name].missing_value=-1.e+034        
+            
+            fid.variables[name].assignValue([[[0.3400644, 0, -46.63519, -6.50198],
+                                              [-0.1214216, 0, 0, 0],
+                                              [0, 0, 0, 0],
+                                              [0, 0, 0, 0]],
+                                             [[0.3400644, 2.291054e-005, -23.33335, -6.50198],
+                                              [-0.1213987, 4.581959e-005, -1.594838e-007, 1.421085e-012],
+                                              [2.291054e-005, 4.582107e-005, 4.581715e-005, 1.854517e-009],
+                                              [0, 2.291054e-005, 2.291054e-005, 0]],
+                                             [[0.3400644, 0.0001374632, -23.31503, -6.50198],
+                                              [-0.1212842, 0.0002756907, 0.006325484, 1.380492e-006],
+                                              [0.0001374632, 0.0002749264, 0.0002742863, 6.665601e-008],
+                                              [0, 0.0001374632, 0.0001374632, 0]],
+                                             [[0.3400644, 0.0002520159, -23.29672, -6.50198],
+                                              [-0.1211696, 0.0005075303, 0.01264618, 6.208276e-006],
+                                              [0.0002520159, 0.0005040318, 0.0005027961, 2.23865e-007],
+                                              [0, 0.0002520159, 0.0002520159, 0]],
+                                             [[0.3400644, 0.0003665686, -23.27842, -6.50198],
+                                              [-0.1210551, 0.0007413362, 0.01896192, 1.447638e-005],
+                                              [0.0003665686, 0.0007331371, 0.0007313463, 4.734126e-007],
+                                              [0, 0.0003665686, 0.0003665686, 0]],
+                                             [[0.3400644, 0.0004811212, -23.26012, -6.50198],
+                                              [-0.1209405, 0.0009771062, 0.02527271, 2.617787e-005],
+                                              [0.0004811212, 0.0009622425, 0.0009599366, 8.152277e-007],
+                                              [0, 0.0004811212, 0.0004811212, 0]]])
             
         
@@ -1862 +1868 @@
 
 
-    def test_ferret2sww(self):
+    def test_ferret2sww1(self):
         """Test that georeferencing etc works when converting from
         ferret format (lat/lon) to sww format (UTM)
@@ -1881 +1887 @@
         #Read
         from coordinate_transforms.redfearn import redfearn
-        fid = NetCDFFile(self.test_MOST_file)
+        #fid = NetCDFFile(self.test_MOST_file)
+        fid = NetCDFFile(self.test_MOST_file + '_ha.nc')        
         first_value = fid.variables['HA'][:][0,0,0]
         fourth_value = fid.variables['HA'][:][0,0,3]
@@ -1887 +1894 @@
 
         #Call conversion (with zero origin)
-        ferret2sww('small', verbose=False,
+        #ferret2sww('small', verbose=False,
+        #           origin = (56, 0, 0))
+        ferret2sww(self.test_MOST_file, verbose=False,
                    origin = (56, 0, 0))
-
 
         #Work out the UTM coordinates for first point
@@ -1896 +1904 @@
 
         #Read output file 'small.sww'
-        fid = NetCDFFile('small.sww')
+        #fid = NetCDFFile('small.sww')
+        fid = NetCDFFile(self.test_MOST_file + '.sww')        
 
         x = fid.variables['x'][:]
@@ -1921 +1930 @@
         #Cleanup
         import os
-        os.remove('small.sww')
-
+        #os.remove('small.sww')
+        os.remove(self.test_MOST_file + '.sww')
 
 
@@ -1943 +1952 @@
 
         #fid = NetCDFFile('small_ha.nc')
-        fid = NetCDFFile(self.test_MOST_file)
+        fid = NetCDFFile(self.test_MOST_file + '_ha.nc')
 
         #Pick a coordinate and a value
@@ -1960 +1969 @@
 
         #Call conversion (with zero origin)
-        ferret2sww('small', verbose=False,
+        ferret2sww(self.test_MOST_file, verbose=False,
                    origin = (56, 0, 0))
 
@@ -1968 +1977 @@
 
         #Read output file 'small.sww'
-        fid = NetCDFFile('small.sww')
+        fid = NetCDFFile(self.test_MOST_file + '.sww')
 
         x = fid.variables['x'][:]
@@ -1986 +1995 @@
         #Cleanup
         import os
-        os.remove('small.sww')
+        os.remove(self.test_MOST_file + '.sww')
 
 
@@ -2154 +2163 @@
 
         #Call conversion (with nonzero origin)
-        ferret2sww('small', verbose=False,
+        ferret2sww(self.test_MOST_file, verbose=False,
                    origin = (56, 100000, 200000))
 
@@ -2162 +2171 @@
 
         #Read output file 'small.sww'
-        fid = NetCDFFile('small.sww', 'r')
-
+        #fid = NetCDFFile('small.sww', 'r')
+        fid = NetCDFFile(self.test_MOST_file + '.sww')
+        
         x = fid.variables['x'][:]
         y = fid.variables['y'][:]
@@ -2174 +2184 @@
 
         #Cleanup
-        os.remove('small.sww')
+        os.remove(self.test_MOST_file + '.sww')