def testGetTag(self):
tagName = 'someGreatTag'
tagAttr = nodeTag.getTagAttr(tagName)
self.assertRaises(RuntimeError, nodeTag.getTag, self.xform, tagName)
self.assertEqual(nodeTag.getTag(self.xform, tagName, noError=True), {})
val = {'test': 'x'}
nodeTag.setTag(self.xform, tagName, val)
self.assertEqual(nodeTag.getTag(self.xform, tagName), val)
def testHasTag(self):
tagName = 'someGreatTag'
tagAttr = nodeTag.getTagAttr(tagName)
self.assertFalse(nodeTag.hasTag(self.xform,tagName))
self.assertFalse(nodeTag.hasTag(self.xform,tagAttr))
nodeTag.setTag(self.xform, tagName, {})
self.assertTrue(nodeTag.hasTag(self.xform,tagName))
self.assertTrue(nodeTag.hasTag(self.xform,tagAttr))
def testValidTags(self):
tagName = 'someGreatTag'
tagAttr = nodeTag.getTagAttr(tagName)
validValues = {'string': 'aStr',
'int': 5,
'float': 5.5,
'list': ['x', 5, 5.5],
'dict': {'s': 'x', 'i': 5, 'f': 5.5},
'tuple': ('x', 5, 5.5),
'set': set(['x', 5, 5.5])}
nodeTag.setTag(self.xform, tagName, validValues)
gottenTag = nodeTag.getTag(self.xform, tagName)
for k, v in validValues.items():
self.assertEqual(v, gottenTag[k])
def testSetTag(self):
tagName = 'someGreatTag'
tagAttr = nodeTag.getTagAttr(tagName)
nodeTag.setTag(self.xform, tagName, {})
self.assertTrue(MC.attributeQuery(tagAttr, n=self.xform, ex=1))
def testTagPrefix(self):
tagName = 'someGreatTag'
self.assertEqual(nodeTag.getTagAttr(tagName), 'beingsTag_someGreatTag')
tagName = 'beingsTag_someGreatTag'
self.assertEqual(nodeTag.getTagAttr(tagName), 'beingsTag_someGreatTag')
