def setUp(self): self.t = Tree(RootNode('root', [0, 0, 0], []), []) self.t.add_child('gate1', RootNode('gate1', [1, 2, 3], [])) self.t.add_child('gate11', RootNode('gate11', [4, 5, 6], [])) self.t.visit('root') self.t.add_child('gate2', RootNode('gate2', [2, 3, 4], [])) self.t.add_child('gate21', RootNode('gate21', [3, 4, 5], [])) self.t.add_child('gate211', RootNode('gate211', [4, 5, 6], []))
class TreeTestCase(unittest.TestCase): def setUp(self): self.t = Tree(RootNode('root', [0, 0, 0], []), []) self.t.add_child('gate1', RootNode('gate1', [1, 2, 3], [])) self.t.add_child('gate11', RootNode('gate11', [4, 5, 6], [])) self.t.visit('root') self.t.add_child('gate2', RootNode('gate2', [2, 3, 4], [])) self.t.add_child('gate21', RootNode('gate21', [3, 4, 5], [])) self.t.add_child('gate211', RootNode('gate211', [4, 5, 6], [])) def testView(self): assert self.t.view() == [4, 5, 6], 'view current node failed' def testRename(self): self.t.rename_node('gate1', 'foo') self.t.visit('foo') assert self.t.get().name == 'foo', 'rename failed to change node name' assert self.t.view() == [1, 2, 3], 'rename gate1 to foo failed' self.assertRaises(KeyError, self.t.rename_node, 'foo', 'gate2') def testVisitError(self): self.t.visit('root') self.assertRaises(KeyError, self.t.visit, 2) self.assertRaises(KeyError, self.t.visit, 'this node does not exist')