Exemplo n.º 1
0
    def test_delete_child(self):
        test_parent_node = UnorderedNodeBase(node_name='parent_node')
        test_child_node_1 = UnorderedNodeBase(node_name='child_node_1')
        test_child_node_2 = UnorderedNodeBase(node_name='child_node_2')
        test_parent_node.add_child(child_node=test_child_node_1)
        test_parent_node.add_child(child_node=test_child_node_2)

        test_parent_node.delete_child(child_node=test_child_node_1)
        self.assertEqual(test_parent_node.get_num_children(), 1)

        test_parent_node.delete_child(child_node=test_child_node_2)
        self.assertEqual(test_parent_node.get_num_children(), 0)
Exemplo n.º 2
0
    def test_combined_tests_3(self):
        test_child_node_1 = UnorderedNodeBase(node_name='child_node_1')
        test_child_node_2 = UnorderedNodeBase(node_name='child_node_2')
        test_parent_node_1 = UnorderedNodeBase(node_name='parent_node_1')
        test_parent_node_2 = UnorderedNodeBase(node_name='parent_node_2')

        test_parent_node_1.add_child(child_node=test_child_node_1)
        test_parent_node_2.add_child(child_node=test_child_node_1)
        test_parent_node_1.add_child(child_node=test_child_node_2)
        test_parent_node_2.add_child(child_node=test_child_node_2)

        test_parent_node_1.delete_child(child_node=test_child_node_1)

        self.assertSetEqual(set(test_parent_node_1.get_children_names()), {'child_node_2'})
        self.assertSetEqual(set(test_parent_node_2.get_children_names()), {'child_node_1', 'child_node_2'})
        self.assertSetEqual(set(test_child_node_1.get_parents_names()), {'parent_node_2'})
        self.assertSetEqual(set(test_child_node_2.get_parents_names()), {'parent_node_1', 'parent_node_2'})