Changeset 281 for inundation/ga/storm_surge/pyvolution/test_data_manager.py

Timestamp:

Sep 7, 2004, 10:29:42 PM (21 years ago)

Author:

ole

Message:

Got datamanager for sww smooth to work.
Non-smooth does not work yet and tests are unfinished

File:

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

inundation/ga/storm_surge/pyvolution/test_data_manager.py

@@ -128 +128 @@
 
 
-#     def test_sww_constant(self):
-#         """Test that constant sww information can be written correctly
-#         (non smooth)
-#         """
-
-#         import time, os
-#         from Numeric import array, zeros, allclose, Float, concatenate
-#         from Scientific.IO.NetCDF import NetCDFFile        
-
-#         self.domain.filename = 'datatest' + str(time.time())
-#         self.domain.format = 'sww'
-#         self.domain.smooth = False
-        
-#         sww = get_dataobject(self.domain)
-#         sww.store_all()
-
-#         #Check contents
-#         #Get NetCDF
-#         fid = NetCDFFile(sww.filename, 'r')  #Open existing file for append
-        
-#         # Get the variables
-#         x = fid.variables['x']
-#         y = fid.variables['y']
-#         z = fid.variables['z']
-
-#         volumes = fid.variables['volumes']        
-
-#         assert allclose (x[:], concatenate( (self.X0, self.X1, self.X2) ))
-#         assert allclose (y[:], concatenate( (self.Y0, self.Y1, self.Y2) ))
-#         assert allclose (z[:], concatenate( (self.F0[:,0],
-#                                              self.F1[:,0],
-#                                              self.F2[:,0]) ))
-
-#         V = volumes
-
-#         P = len(self.domain)
-#         for k in range(P):
-#              assert V[k,0] == V[k,1] - P
-#              assert V[k,1] == V[k,2] - P
-        
-
-
-        
-#         fid.close()
-        
-#         #Cleanup
-#         os.remove(sww.filename)
-
-
-#     def test_sww_constant_smooth(self):
-#         """Test that constant sww information can be written correctly
-#         (non smooth)
-#         """
-
-
-#         import time, os
-#         from Numeric import array, zeros, allclose, Float, concatenate
-#         from Scientific.IO.NetCDF import NetCDFFile        
-
-#         self.domain.filename = 'datatest' + str(time.time())
-#         self.domain.format = 'sww'
-#         self.domain.smooth = True
-        
-#         sww = get_dataobject(self.domain)
-#         sww.store_all()
-
-#         #Check contents
-#         #Get NetCDF
-#         fid = NetCDFFile(sww.filename, 'r')  #Open existing file for append
-        
-#         # Get the variables
-#         x = fid.variables['x']
-#         y = fid.variables['y']
-#         z = fid.variables['z']
-
-#         volumes = fid.variables['volumes']        
-
-#         X = x[:]
-#         Y = y[:]
-        
-#         assert allclose([X[0], Y[0]], array([0.0, 0.0]))
-#         assert allclose([X[1], Y[1]], array([0.0, 0.5]))
-#         assert allclose([X[2], Y[2]], array([0.0, 1.0]))
-
-#         assert allclose([X[4], Y[4]], array([0.5, 0.5]))
-
-#         assert allclose([X[7], Y[7]], array([1.0, 0.5]))        
-
-#         Z = z[:]
-#         assert Z[4] == -0.5
-
-
-
-#         V = volumes
-#         assert V[2,0] == 4
-#         assert V[2,1] == 5
-#         assert V[2,2] == 1
-
-#         assert V[4,0] == 6
-#         assert V[4,1] == 7
-#         assert V[4,2] == 3                      
-        
-        
-
-
-        
-#         fid.close()
-        
-#         #Cleanup
-#         os.remove(sww.filename)
-
-
-
-#     def test_sww_variable(self):
-#         """Test that sww information can be written correctly
-#         """
-
-#         import time, os
-#         from Numeric import array, zeros, allclose, Float, concatenate
-#         from Scientific.IO.NetCDF import NetCDFFile        
-
-#         self.domain.filename = 'datatest' + str(time.time())
-#         self.domain.format = 'sww'
-#         self.domain.smooth = True
-#         self.domain.reduction = mean             
-        
-#         sww = get_dataobject(self.domain)
-#         sww.store_all()
-#         sww.store_timestep()
-
-#         #Check contents
-#         #Get NetCDF
-#         fid = NetCDFFile(sww.filename, 'r')  #Open existing file for append
-        
-
-#         # Get the variables
-#         x = fid.variables['x']
-#         y = fid.variables['y']
-#         z = fid.variables['z']        
-#         time = fid.variables['time']
-#         stage = fid.variables['stage']
-
-
-#         Q0 = self.domain.volume_class.conserved_quantities_vertex0
-#         Q1 = self.domain.volume_class.conserved_quantities_vertex1
-#         Q2 = self.domain.volume_class.conserved_quantities_vertex2        
-
-#         A = stage[0,:]
-#         #print A[0], (Q2[0,0] + Q1[1,0])/2
-#         assert allclose(A[0], (Q2[0,0] + Q1[1,0])/2)  
-#         assert allclose(A[1], (Q0[1,0] + Q1[3,0] + Q2[2,0])/3)
-#         assert allclose(A[2], Q0[3,0])
-#         assert allclose(A[3], (Q0[0,0] + Q1[5,0] + Q2[4,0])/3)
-
-#         #Center point
-#         assert allclose(A[4], (Q1[0,0] + Q2[1,0] + Q0[2,0] +\
-#                                  Q0[5,0] + Q2[6,0] + Q1[7,0])/6)
-                                 
-        
-        
-#         fid.close()
-        
-#         #Cleanup
-#         os.remove(sww.filename)
+    def test_sww_constant(self):
+        """Test that constant sww information can be written correctly
+        (non smooth)
+        """
+
+        import time, os
+        from Numeric import array, zeros, allclose, Float, concatenate
+        from Scientific.IO.NetCDF import NetCDFFile        
+
+        self.domain.filename = 'datatest' + str(time.time())
+        self.domain.format = 'sww'
+        self.domain.smooth = False
+      
+        sww = get_dataobject(self.domain)
+        sww.store_connectivity()
+
+        #Check contents
+        #Get NetCDF
+        fid = NetCDFFile(sww.filename, 'r')  #Open existing file for append
+      
+        # Get the variables
+        x = fid.variables['x']
+        y = fid.variables['y']
+        z = fid.variables['z']
+
+        volumes = fid.variables['volumes']        
+
+        #print z[:]
+        #print self.F.flat
+        #print concatenate( (self.X0, self.X1, self.X2))
+                        
+        assert allclose (x[:], concatenate( (self.X0, self.X1, self.X2) ))
+        assert allclose (y[:], concatenate( (self.Y0, self.Y1, self.Y2) ))
+        assert allclose (z[:], self.F.flat)
+
+        V = volumes
+
+        P = len(self.domain)
+        for k in range(P):
+             assert V[k,0] == V[k,1] - P
+             assert V[k,1] == V[k,2] - P
+      
+
+
+      
+        fid.close()
+      
+        #Cleanup
+        os.remove(sww.filename)
+
+
+    def test_sww_constant_smooth(self):
+        """Test that constant sww information can be written correctly
+        (non smooth)
+        """
+
+
+        import time, os
+        from Numeric import array, zeros, allclose, Float, concatenate
+        from Scientific.IO.NetCDF import NetCDFFile        
+
+        self.domain.filename = 'datatest' + str(time.time())
+        self.domain.format = 'sww'
+        self.domain.smooth = True
+      
+        sww = get_dataobject(self.domain)
+        sww.store_connectivity()        
+
+        #Check contents
+        #Get NetCDF
+        fid = NetCDFFile(sww.filename, 'r')  #Open existing file for append
+      
+        # Get the variables
+        x = fid.variables['x']
+        y = fid.variables['y']
+        z = fid.variables['z']
+
+        volumes = fid.variables['volumes']        
+
+        X = x[:]
+        Y = y[:]
+      
+        assert allclose([X[0], Y[0]], array([0.0, 0.0]))
+        assert allclose([X[1], Y[1]], array([0.0, 0.5]))
+        assert allclose([X[2], Y[2]], array([0.0, 1.0]))
+
+        assert allclose([X[4], Y[4]], array([0.5, 0.5]))
+
+        assert allclose([X[7], Y[7]], array([1.0, 0.5]))        
+
+        Z = z[:]
+        assert Z[4] == -0.5
+
+        V = volumes
+        assert V[2,0] == 4
+        assert V[2,1] == 5
+        assert V[2,2] == 1
+
+        assert V[4,0] == 6
+        assert V[4,1] == 7
+        assert V[4,2] == 3                      
+      
+      
+
+
+      
+        fid.close()
+      
+        #Cleanup
+        os.remove(sww.filename)
+
+
+
+    def test_sww_variable(self):
+        """Test that sww information can be written correctly
+        """
+
+        import time, os
+        from Numeric import array, zeros, allclose, Float, concatenate
+        from Scientific.IO.NetCDF import NetCDFFile        
+
+        self.domain.filename = 'datatest' + str(time.time())
+        self.domain.format = 'sww'
+        self.domain.smooth = True
+        self.domain.reduction = mean             
+      
+        sww = get_dataobject(self.domain)
+        sww.store_connectivity()
+        sww.store_timestep('level')
+
+        #Check contents
+        #Get NetCDF
+        fid = NetCDFFile(sww.filename, 'r')  #Open existing file for append
+      
+
+        # Get the variables
+        x = fid.variables['x']
+        y = fid.variables['y']
+        z = fid.variables['z']        
+        time = fid.variables['time']
+        stage = fid.variables['stage']
+
+
+        Q = self.domain.quantities['level']     
+        Q0 = Q.vertex_values[:,0]
+        Q1 = Q.vertex_values[:,1]
+        Q2 = Q.vertex_values[:,2]
+
+        A = stage[0,:]
+        #print A[0], (Q2[0,0] + Q1[1,0])/2
+        assert allclose(A[0], (Q2[0] + Q1[1])/2)  
+        assert allclose(A[1], (Q0[1] + Q1[3] + Q2[2])/3)
+        assert allclose(A[2], Q0[3])
+        assert allclose(A[3], (Q0[0] + Q1[5] + Q2[4])/3)
+
+        #Center point
+        assert allclose(A[4], (Q1[0] + Q2[1] + Q0[2] +\
+                                 Q0[5] + Q2[6] + Q1[7])/6)
+                               
+      
+      
+        fid.close()
+      
+        #Cleanup
+        os.remove(sww.filename)
         
         
@@ -421 +422 @@
 #-------------------------------------------------------------
 if __name__ == "__main__":
-    #suite = unittest.makeSuite(dataTestCase,'test_sww_constant')
-    suite = unittest.makeSuite(dataTestCase,'test_dummy')
+    suite = unittest.makeSuite(dataTestCase,'test')
     runner = unittest.TextTestRunner()
     runner.run(suite)