def test_triangle(self): node1 = tree.TreeNode(1) node2 = tree.TreeNode(4) node3 = tree.TreeNode(6) tree.link(node1, node2) tree.link(node1, node3) self.assertEqual(node1, tree.DepthFirstSearch(node1, 1)) self.assertEqual(node2, tree.DepthFirstSearch(node1, 4)) self.assertEqual(node3, tree.DepthFirstSearch(node1, 6)) self.assertEqual(None, tree.DepthFirstSearch(node1, 7))
def test_link(self): node1 = tree.TreeNode(2) node2 = tree.TreeNode(3) node3 = tree.TreeNode(5) tree.link(node1, node2) tree.link(node1, node3) self.assertEqual(None, node1.parent) self.assertEqual(node2, node1.children[0]) self.assertEqual(node1, node2.parent) self.assertEqual([], node2.children) self.assertEqual(node3, node1.children[1]) self.assertEqual(node1, node3.parent) self.assertEqual([], node3.children)