def test_get_path_for_success(self):
child_1 = heat_gen.TreeNode()
self.tree.add_child(child_1)
child_2 = heat_gen.TreeNode()
child_1.add_child(child_2)
child_3 = heat_gen.TreeNode()
child_2.add_child(child_3)
path = [self.tree, child_1, child_2, child_3]
self.assertListEqual(path, child_3.get_path())
def test_get_leaves_for_success(self):
child_1 = heat_gen.TreeNode()
self.tree.add_child(child_1)
child_2 = heat_gen.TreeNode()
child_1.add_child(child_2)
child_3 = heat_gen.TreeNode()
child_2.add_child(child_3)
child_4 = heat_gen.TreeNode()
child_2.add_child(child_4)
child_5 = heat_gen.TreeNode()
child_2.add_child(child_5)
leaves = [child_3, child_4, child_5]
self.assertListEqual(leaves, heat_gen.TreeNode.get_leaves(self.tree))
def test_get_children_for_success(self):
self.assertListEqual(list(), self.tree.get_children())
child = heat_gen.TreeNode()
self.tree.add_child(child)
children = [child]
self.assertListEqual(children, self.tree.get_children())
def test_get_parent_for_success(self):
self.assertIsNone(self.tree.get_parent())
child = heat_gen.TreeNode()
self.tree.add_child(child)
self.assertEqual(self.tree, child.get_parent())
def test_add_child_for_success(self):
child = heat_gen.TreeNode()
self.tree.add_child(child)
self.assertIn(child, self.tree.down)
def setUp(self):
self.tree = heat_gen.TreeNode()