예제 #1
0
 def test_flyTags(self):
     dom = ReadMesh(os.path.join(DUDLEY_TEST_MESH_PATH, "tagtest2.fly"))
     tags = sorted(dom.showTagNames().split(', '))
     self.assertEqual(tags, sorted(['tag1', 'tag2', 'tag3', 'tag4', 'All']))
     self.assertEqual(dom.getTag('tag1'), 5, 'error with tag1')
     self.assertEqual(dom.getTag('tag2'), 8, 'error with tag2,')
     self.assertEqual(dom.getTag('tag3'), 6, 'error with tag3')
     self.assertEqual(dom.getTag('tag4'), 7, 'error with tag4')
     self.assertEqual(dom.getTag('All'), 10, 'error with All')
     self.assertRaises(ValueError, dom.getTag, 'tag6')
예제 #2
0
 def test_ReadWriteTagNames(self):
     file="hex_2D_order2.msh"
     test = os.path.join(DUDLEY_WORKDIR,"test.fly")
     dom = ReadMesh(os.path.join(DUDLEY_TEST_MESH_PATH,file),3,optimize=False)
     insertTagNames(dom,A=1,B=2)
     dom.write(test)
     dom2 = ReadMesh(test,3,optimize=False)
     t=getTagNames(dom)
     self.assertTrue(len(t)==6)
     self.assertTrue("A" in t)
     self.assertTrue("B" in t)
     self.assertTrue(dom2.getTag("A") == 1)
     self.assertTrue(dom2.getTag("B") == 2)
     self.assertTrue(dom2.isValidTagName("A"))
     self.assertTrue(dom2.isValidTagName("B"))