Changeset 546 for inundation/ga/storm_surge/pyvolution/test_domain.py

Timestamp:

Nov 16, 2004, 10:40:37 AM (20 years ago)

Author:

duncan

Message:

geting and setting values with a set of triangles - used with region tags

File:

• inundation/ga/storm_surge/pyvolution/test_domain.py (modified) (3 diffs)

inundation/ga/storm_surge/pyvolution/test_domain.py

@@ -12 +12 @@
     if tag == "mound":
         domain.test = "Mound"
-        
-
-
+
+
+def set_bottom_friction(tag, elements, domain):
+    if tag == "bottom":
+        #print 'bottom - indexes',elements
+        domain.set_quantity('friction', 0.09, indexes = elements)
+ 
+def set_top_friction(tag, elements, domain):
+    if tag == "top":
+        #print 'top - indexes',elements
+        domain.set_quantity('friction', 1., indexes = elements)
+        
+ 
+def set_all_friction(tag, elements, domain):
+    if tag == "all":
+        new_values = domain.get_quantity('friction', indexes = elements) + 10.0
+        
+        domain.set_quantity('friction', new_values, indexes = elements)
+                  
+      
 class TestCase(unittest.TestCase):
     def setUp(self):
@@ -276 +293 @@
 
 
-    def testz_set_region(self):
+    def test_set_region(self):
         """Set quantities for sub region
         """
@@ -330 +347 @@
                         'set region failed')
         
+
+
+    def test_region_tags(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([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]])
 #-------------------------------------------------------------
 if __name__ == "__main__":