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')
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"))