Changeset 1018 for inundation/ga/storm_surge/pyvolution/test_cg_solve.py

Timestamp:

Mar 7, 2005, 9:32:07 AM (20 years ago)

Author:

steve

Message:

Cleaned up test function

File:

• inundation/ga/storm_surge/pyvolution/test_cg_solve.py (modified) (11 diffs)

inundation/ga/storm_surge/pyvolution/test_cg_solve.py

@@ -9 +9 @@
 
 
-class TestCase(unittest.TestCase):
+class Test_CG_Solve(unittest.TestCase):
 
     def test_sparse_solve(self):
         """Small Sparse Matrix"""
-        
+
         A = [[2.0, -1.0, 0.0, 0.0 ],
              [-1.0, 2.0, -1.0, 0.0],
              [0.0, -1.0, 2.0, -1.0],
              [0.0,0.0, -1.0, 2.0]]
-        
+
         A = Sparse(A)
 
@@ -35 +35 @@
         n = 50
         A = Sparse(n,n)
-        
+
         for i in arange(0,n):
             A[i,i] = 1.0
@@ -42 +42 @@
             if i < n-1 :
                 A[i,i+1] = -0.5
-                
+
         xe = ones( (n,), Float)
 
@@ -52 +52 @@
     def test_solve_large_2d(self):
         """Standard 2d laplacian"""
-        
+
         n = 20
         m = 10
@@ -70 +70 @@
                 if j < m-1 :
                     A[I,I+1] = -1.0
-                
+
         xe = ones( (n*m,), Float)
 
@@ -81 +81 @@
         """Standard 2d laplacian with csr format
         """
-        
+
         n = 100
         m = 100
@@ -99 +99 @@
                 if j < m-1 :
                     A[I,I+1] = -1.0
-                
+
         xe = ones( (n*m,), Float)
 
@@ -114 +114 @@
     def test_solve_large_2d_with_default_guess(self):
         """Standard 2d laplacian using default first guess"""
-        
+
         n = 20
         m = 10
@@ -132 +132 @@
                 if j < m-1 :
                     A[I,I+1] = -1.0
-                
+
         xe = ones( (n*m,), Float)
 
@@ -143 +143 @@
     def test_vector_shape_error(self):
         """Raise VectorShapeError"""
-        
+
         A = [[2.0, -1.0, 0.0, 0.0 ],
              [-1.0, 2.0, -1.0, 0.0],
              [0.0, -1.0, 2.0, -1.0],
              [0.0,0.0, -1.0, 2.0]]
-        
+
         A = Sparse(A)
 
@@ -161 +161 @@
             raise msg
 
-        
+
 #-------------------------------------------------------------
 if __name__ == "__main__":
-     suite = unittest.makeSuite(TestCase,'test')
+     suite = unittest.makeSuite(Test_CG_Solve,'test')
      runner = unittest.TextTestRunner() #(verbosity=2)
      runner.run(suite)
 
-    
-