Changeset 5846 for anuga_core/source/pymetis

Timestamp:

Oct 18, 2008, 7:52:32 AM (16 years ago)

Author:

steve

Message:

Added test to test_metis.py

File:

• anuga_core/source/pymetis/test_metis.py (modified) (4 diffs)

anuga_core/source/pymetis/test_metis.py

@@ -3 +3 @@
 from sys import path
 
-from Numeric import array
+from Numeric import array, allclose
 
 import unittest
@@ -9 +9 @@
 # path.append('..' + sep + 'pymetis')
 
-import metis
+import pymetis.metis  as metis
 
 class TestMetis(unittest.TestCase):
@@ -15 +15 @@
         pass
 
-    def testHexmesh(self):
+    def test_Hexmesh(self):
         # Hexagonal mesh
         #
@@ -42 +42 @@
         npart_expected = array([0, 2, 2, 2, 0, 0, 0], 'i')
         self.assert_(edgecut == 5)
-        for i in range(len(epart)):
-            self.assert_(epart[i] == epart_expected[i])
-        for i in range(len(npart)):
-            self.assert_(npart[i] == npart_expected[i])
+        assert allclose(epart, epart_expected)
+        assert allclose(npart, npart_expected)        
+
+
+    def test_Hexmesh2(self):
+        # Hexagonal mesh
+        #
+        #   1---2  
+        #  / \ / \
+        # 6---0---3
+        #  \ / \ /
+        #   5---4
+        #
+        # Divided 2 ways
+        # Calling order is: elements, verticies, edge list
+        # element type, number parts
+        edgecut, epart, npart = metis.partMeshNodal(6, 7,\
+                                                    [0, 2, 1,\
+                                                     0, 3, 2,\
+                                                     0, 4, 3,\
+                                                     0, 5, 4,\
+                                                     0, 6, 5,\
+                                                     0, 1, 6],\
+                                                    1,\
+                                                    2,)
+        #print edgecut
+        #print epart
+        #print npart
+        epart_expected = array([1, 0, 0, 0, 1, 1], 'i')
+        npart_expected = array([0, 1, 1, 0, 0, 0, 1], 'i')
+        self.assert_(edgecut == 5)
+        assert allclose(epart, epart_expected)
+        assert allclose(npart, npart_expected)
+
 
 if __name__ == '__main__':