Changeset 2492 for inundation/pyvolution/test_data_manager.py

Timestamp:

Mar 8, 2006, 2:23:18 PM (19 years ago)

Author:

sexton

Message:

(1) Updating dem2pts to remove NODATA_values (2) accompanying unit test

File:

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

inundation/pyvolution/test_data_manager.py

@@ -764 +764 @@
         os.remove(root + '.prj')
 
+    def test_dem2pts_remove_Nullvalues(self):
+        """Test conversion from dem in ascii format to native NetCDF xya format
+        """
+
+        import time, os
+        from Numeric import array, zeros, allclose, Float, concatenate
+        from Scientific.IO.NetCDF import NetCDFFile
+
+        #Write test asc file
+        root = 'demtest'
+
+        filename = root+'.asc'
+        fid = open(filename, 'w')
+        fid.write("""ncols         5
+nrows         6
+xllcorner     2000.5
+yllcorner     3000.5
+cellsize      25
+NODATA_value  -9999
+""")
+        #Create linear function
+        # ref_ will write all the values
+        # new_ref_ will write the values except for NODATA_values
+        ref_points = []
+        ref_elevation = []
+        new_ref_pts = []
+        new_ref_elev = []
+        NODATA_value = -9999
+        for i in range(6):
+            y = (6-i)*25.0
+            for j in range(5):
+                x = j*25.0
+                z = x+2*y
+                #if j == 4: z = NODATA_value
+                if i == 1 and j == 2: z = NODATA_value
+                if i == 2 and j == 4: z = NODATA_value
+                if i == 3 and j == 3: z = NODATA_value
+                if i == 5 and j == 1: z = NODATA_value
+                
+                if z <> NODATA_value:
+                    new_ref_elev.append(z)
+                    new_ref_pts.append( [x,y] )
+                
+                ref_points.append( [x,y] )
+                ref_elevation.append(z)
+
+                fid.write('%f ' %z)
+            fid.write('\n')
+
+        fid.close()
+
+        nn = ref_elevation.count(NODATA_value)
+        
+        #Write prj file with metadata
+        metafilename = root+'.prj'
+        fid = open(metafilename, 'w')
+
+
+        fid.write("""Projection UTM
+Zone 56
+Datum WGS84
+Zunits NO
+Units METERS
+Spheroid WGS84
+Xshift 0.0000000000
+Yshift 10000000.0000000000
+Parameters
+""")
+        fid.close()
+
+        #Convert to NetCDF pts
+        convert_dem_from_ascii2netcdf(root)
+        dem2pts(root)
+
+        #Check contents
+        #Get NetCDF
+        fid = NetCDFFile(root+'.pts', 'r')
+
+        # Get the variables
+        #print fid.variables.keys()
+        points = fid.variables['points']
+        elevation = fid.variables['elevation']
+
+        #Check values
+
+        assert len(points) == len(new_ref_pts), 'length of returned points not correct'
+        assert allclose(points, new_ref_pts), 'points do not align'
+
+        assert len(elevation) == len(new_ref_elev), 'length of returned elevation not correct'
+        assert allclose(elevation, new_ref_elev), 'elevations do not align'
+
+        #Cleanup
+        fid.close()
+
+
+        os.remove(root + '.pts')
+        os.remove(root + '.dem')
+        os.remove(root + '.asc')
+        os.remove(root + '.prj')
+
+
+