def test_delete_node_subtree(self):
root = SimpleTreeNode(0, None)
tree = SimpleTree(root)
node = SimpleTreeNode(1, None)
node2 = SimpleTreeNode(2, None)
node3 = SimpleTreeNode(3, None)
node4 = SimpleTreeNode(1, None)
node5 = SimpleTreeNode(5, None)
node6 = SimpleTreeNode(1, None)
tree.AddChild(root, node)
tree.AddChild(root, node2)
tree.AddChild(root, node3)
tree.AddChild(node3, node4)
tree.AddChild(node3, node5)
tree.AddChild(node3, node6)
self.assertEqual(tree.Count(), 7)
tree.DeleteNode(node3)
self.assertEqual(tree.Count(), 3)
for item in [node3, node4, node5, node6]:
self.assertNotIn(item, tree.GetAllNodes())
def test_delete_node(self):
root = SimpleTreeNode(0, None)
tree = SimpleTree(root)
node = SimpleTreeNode(1, None)
node2 = SimpleTreeNode(2, None)
tree.AddChild(root, node)
tree.AddChild(root, node2)
self.assertEqual(tree.Count(), 3)
tree.DeleteNode(node)
self.assertEqual(tree.Count(), 2)
self.assertNotIn(node, tree.GetAllNodes())
def test_get_all_nodes(self):
root = SimpleTreeNode(0, None)
tree = SimpleTree(root)
node = SimpleTreeNode(1, None)
node2 = SimpleTreeNode(2, None)
expected_nodes = [root, node, node2]
tree.AddChild(root, node)
tree.AddChild(root, node2)
nodes = tree.GetAllNodes()
self.assertEqual(len(nodes), len(expected_nodes))
for item in expected_nodes:
self.assertIn(item, nodes)
def test_get_all_nodes_empty_tree(self):
tree = SimpleTree(None)
self.assertEqual(tree.GetAllNodes(), [])