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)
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'})