def get_childs_depth1(self):
"""Test to retrieve all childs and their path on
test_add_child_depth2_node_already_exist tree
"""
root = Node('A')
root.add_leaf(['A', 'B', 'C'], 1)
root.add_leaf(['A', 'B', 'D'], 2)
childs = root.get_childs()
childs.sort()
assert childs == [(['A', 'B', 'C'], 1), (['A', 'B', 'C'], 2)]
def test_retrieve_2_leaf_value_depth2(self):
"""
Test to retrieve the two leafs added in
test_add_child_depth2_node_already_exist
"""
root = Node('A')
root.add_leaf(['A', 'B', 'C'], 1)
root.add_leaf(['A', 'B', 'D'], 2)
leafs = root.retrieve_leaf_values(['A', 'B'])
leafs.sort()
assert leafs == [1, 2]
def test_add_child_depth2(self):
"""Test to add a leaf to a tree of depth 2"""
root = Node('A')
root.add_leaf(['A', 'B', 'C'], 1)
assert len(root.childs) == 1
node_b = root.childs[0]
assert node_b.label == 'B'
assert len(node_b.childs) == 1
leaf = node_b.childs[0]
assert leaf == Node('C', 1)
def test_add_child_to_empty_tree(self):
tree = Tree()
tree.add(['A', 'B'], 1)
assert tree.root is not None
assert len(tree.root.childs) == 1
assert tree.root.label == 'A'
assert tree.root.childs[0] == Node('B', 1)
def test_retrieve_1_leaf_value_depth2_2_branch(self):
"""
Test to retrieve the one leafs added in
test_add_child_depth2_node_already_exist
"""
root = Node('A')
root.add_leaf(['A', 'B', 'C'], 1)
root.add_leaf(['A', 'B', 'D'], 2)
[leaf_c] = root.retrieve_leaf_values(['A', 'B', 'C'])
assert leaf_c == 1
[leaf_d] = root.retrieve_leaf_values(['A', 'B', 'D'])
assert leaf_d == 2
def test_add_child_depth2_node_already_exist(self):
"""
Test to add a leaf to a tree of depth 2 when its parent's node already
exists
"""
root = Node('A')
root.add_leaf(['A', 'B', 'C'], 1)
root.add_leaf(['A', 'B', 'D'], 2)
assert len(root.childs) == 1
node_b = root.childs[0]
assert node_b.label == 'B'
assert len(node_b.childs) == 2
sorted_child = deepcopy(node_b.childs)
sorted_child.sort(key=lambda node: node.label)
leaf_c = sorted_child[0]
assert leaf_c == Node('C', 1)
leaf_d = sorted_child[0]
assert leaf_d == Node('C', 1)
def test_get_childs_depth2(self):
"""Test to retrieve all childs and their path on test_add_child_depth2
tree"""
root = Node('A')
root.add_leaf(['A', 'B', 'C'], 1)
assert root.get_childs() == [(['A', 'B', 'C'], 1)]