Changeset 1916 for inundation/pyvolution/test_domain.py

Timestamp:

Oct 14, 2005, 9:46:38 AM (19 years ago)

Author:

ole

Message:

Implemented operator overloading for (pow) in class Quantity and tested.
Wrote create_quantity_from_expression and test.

File:

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

inundation/pyvolution/test_domain.py

@@ -134 +134 @@
         q = domain.get_conserved_quantities(3, edge=2)
         assert allclose(q, [-1.5, -1.5, 0.])
+
+
+
+    def test_create_quantity_from_expression(self):
+        """Quantity created from other quantities using arbitrary expression
+        
+        """
+
+
+        a = [0.0, 0.0]
+        b = [0.0, 2.0]
+        c = [2.0,0.0]
+        d = [0.0, 4.0]
+        e = [2.0, 2.0]
+        f = [4.0,0.0]
+
+        points = [a, b, c, d, e, f]
+        #bac, bce, ecf, dbe, daf, dae
+        vertices = [ [1,0,2], [1,2,4], [4,2,5], [3,1,4]]
+
+        domain = Domain(points, vertices, boundary=None,
+                        conserved_quantities =\
+                        ['stage', 'xmomentum', 'ymomentum'],
+                        other_quantities = ['elevation', 'friction'])
+
+
+        domain.set_quantity('elevation', -1)
+
+        
+        domain.set_quantity('stage', [[1,2,3], [5,5,5],
+                                      [0,0,9], [-6, 3, 3]])
+
+        domain.set_quantity('xmomentum', [[1,2,3], [5,5,5],
+                                          [0,0,9], [-6, 3, 3]])
+
+        domain.set_quantity('ymomentum', [[3,3,3], [4,2,1],
+                                          [2,4,-1], [1, 0, 1]])        
+
+        domain.check_integrity()
+
+
+
+        expression = 'stage - elevation'
+        Q = create_quantity_from_expression(domain, expression)
+
+        assert allclose(Q.vertex_values, [[2,3,4], [6,6,6],
+                                      [1,1,10], [-5, 4, 4]])
+
+        expression = '(xmomentum*xmomentum + ymomentum*ymomentum)**0.5'
+        Q = create_quantity_from_expression(domain, expression)
+
+        X = domain.quantities['xmomentum'].vertex_values
+        Y = domain.quantities['ymomentum'].vertex_values        
+
+        assert allclose(Q.vertex_values, (X**2 + Y**2)**0.5)
+                                      
+
+
+