Changeset 6146 for anuga_core/source/anuga/advection

Timestamp:

Jan 13, 2009, 11:58:10 AM (16 years ago)

Author:

rwilson

Message:

Change Numeric imports to general form - ready to change to NumPy?.

Location:

anuga_core/source/anuga/advection

Files:

• advection.py (modified) (5 diffs)
• test_advection.py (modified) (7 diffs)

anuga_core/source/anuga/advection/advection.py

@@ -32 +32 @@
 
 from anuga.abstract_2d_finite_volumes.domain import *
+
+import Numeric as num
+
+
 Generic_domain = Domain # Rename
 
@@ -66 +70 @@
                                 numproc=numproc)
 
-        import Numeric
         if velocity is None:
-            self.velocity = Numeric.array([1,0],'d')
+            self.velocity = num.array([1,0],'d')
         else:
-            self.velocity = Numeric.array(velocity,'d')
+            self.velocity = num.array(velocity,'d')
 
         #Only first is implemented for advection
@@ -155 +158 @@
 
         import sys
-        from Numeric import zeros, Float
         from anuga.config import max_timestep
 
@@ -228 +230 @@
 
         import sys
-        from Numeric import zeros, Float
         from anuga.config import max_timestep
 
@@ -251 +252 @@
         stage_bdry = Stage.boundary_values
 
-        flux = zeros(1, Float) #Work array for summing up fluxes
+        flux = num.zeros(1, num.Float) #Work array for summing up fluxes
 
         #Loop

anuga_core/source/anuga/advection/test_advection.py

@@ -6 +6 @@
 
 from anuga.config import g, epsilon
-from Numeric import allclose, array, zeros, ones, Float
 from anuga.advection.advection import Domain, Transmissive_boundary, Dirichlet_boundary
+
+import Numeric as num
+
 
 class Test_Advection(unittest.TestCase):
@@ -48 +50 @@
 
         #Populate boundary array with dirichlet conditions.
-        domain.neighbours = array([[-1,-2,-3]])
+        domain.neighbours = num.array([[-1,-2,-3]])
         domain.quantities['stage'].boundary_values[:] = 1.0
 
@@ -102 +104 @@
 
         #Populate boundary array with dirichlet conditions.
-        domain.neighbours = array([[-1,-2,-3]])
+        domain.neighbours = num.array([[-1,-2,-3]])
         domain.quantities['stage'].boundary_values[0] = 1.0
 
@@ -111 +113 @@
         domain.compute_fluxes()
         U = domain.quantities['stage'].explicit_update
-        assert allclose(U, 0)
+        assert num.allclose(U, 0)
 
 
@@ -133 +135 @@
 
         #Populate boundary array with dirichlet conditions.
-        domain.neighbours = array([[1,-1,-2], [0,-3,-4]])
+        domain.neighbours = num.array([[1,-1,-2], [0,-3,-4]])
         domain.set_quantity('stage', [1.0, 0.0], location='centroids')
         domain.distribute_to_vertices_and_edges()
@@ -157 +159 @@
         #Boundaries
         T = Transmissive_boundary(domain)
-        D = Dirichlet_boundary(array([3.1415]))
+        D = Dirichlet_boundary(num.array([3.1415]))
 
         domain.set_boundary( {'left': D, 'right': T, 'bottom': T, 'top': T} )
@@ -164 +166 @@
         #Check that the boundary value gets propagated to all elements
         for t in domain.evolve(yieldstep = 0.05, finaltime = 10):
-            if allclose(domain.quantities['stage'].centroid_values, 3.1415):
+            if num.allclose(domain.quantities['stage'].centroid_values, 3.1415):
                 break
 
-        assert allclose(domain.quantities['stage'].centroid_values, 3.1415)
+        assert num.allclose(domain.quantities['stage'].centroid_values, 3.1415)