Changeset 3447 for inundation/pyvolution/test_domain.py

Timestamp:

Aug 2, 2006, 5:14:16 PM (18 years ago)

Author:

duncan

Message:

Making set_region easier to use.

File:

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

inundation/pyvolution/test_domain.py

@@ -536 +536 @@
 
 
+    def test_region_tags2(self):
+        """
+        get values based on triangle lists.
+        """
+        from mesh_factory import rectangular
+        from shallow_water import Domain
+        from Numeric import zeros, Float
+
+        #Create basic mesh
+        points, vertices, boundary = rectangular(1, 3)
+
+        #Create shallow water domain
+        domain = Domain(points, vertices, boundary)
+        domain.build_tagged_elements_dictionary({'bottom':[0,1],
+                                                 'top':[4,5],
+                                                 'all':[0,1,2,3,4,5]})
+
+
+        #Set friction
+        manning = 0.07
+        domain.set_quantity('friction', manning)
+
+        domain.set_region('top', 'friction', 1.0)
+        domain.set_region('bottom', 'friction', 0.09)
+        
+        #print domain.quantities['friction'].get_values()
+        assert allclose(domain.quantities['friction'].get_values(),\
+                        [[ 0.09,  0.09,  0.09],
+                         [ 0.09,  0.09,  0.09],
+                         [ 0.07,  0.07,  0.07],
+                         [ 0.07,  0.07,  0.07],
+                         [ 1.0,  1.0,  1.0],
+                         [ 1.0,  1.0,  1.0]])
+        
+        domain.set_region([set_bottom_friction, set_top_friction])
+        #print domain.quantities['friction'].get_values()
+        assert allclose(domain.quantities['friction'].get_values(),\
+                        [[ 0.09,  0.09,  0.09],
+                         [ 0.09,  0.09,  0.09],
+                         [ 0.07,  0.07,  0.07],
+                         [ 0.07,  0.07,  0.07],
+                         [ 1.0,  1.0,  1.0],
+                         [ 1.0,  1.0,  1.0]])
+
+        domain.set_region([set_all_friction])
+        #print domain.quantities['friction'].get_values()
+        assert allclose(domain.quantities['friction'].get_values(),
+                        [[ 10.09, 10.09, 10.09],
+                         [ 10.09, 10.09, 10.09],
+                         [ 10.07, 10.07, 10.07],
+                         [ 10.07, 10.07, 10.07],
+                         [ 11.0,  11.0,  11.0],
+                         [ 11.0,  11.0,  11.0]])
 
 #-------------------------------------------------------------