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()