def test_get_num_children(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)
self.assertEqual(test_parent_node.get_num_children(), 2)
def test_has_child(self):
test_parent_node = UnorderedNodeBase(node_name='parent_node')
test_child_node = UnorderedNodeBase(node_name='child_node')
test_parent_node.add_child(child_node=test_child_node)
self.assertTrue(test_parent_node.has_child(child_name='child_node'))
self.assertFalse(test_parent_node.has_child(child_name='child_node_0'))
def test_get_children_nodes(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)
self.assertSetEqual(set(test_parent_node.get_children_nodes()), {test_child_node_1, test_child_node_2})
def test_get_child(self):
test_parent_node = UnorderedNodeBase(node_name='parent_node')
test_child_node = UnorderedNodeBase(node_name='child_node')
test_parent_node.add_child(child_node=test_child_node)
self.assertEqual(test_parent_node.get_child(child_name='child_node'), test_child_node)
self.assertEqual(test_parent_node.get_child(child_name='child_node_0'), None)
def test_combined_tests_4(self):
test_node_1 = UnorderedNodeBase(node_name='node_1')
test_node_2 = UnorderedNodeBase(node_name='node_2')
test_node_3 = UnorderedNodeBase(node_name='node_3')
test_node_1.add_child(child_node=test_node_2)
test_node_2.add_child(child_node=test_node_3)
self.assertEqual(test_node_3.get_parent(parent_name='node_2').get_parent(parent_name='node_1'), test_node_1)
def test_get_children(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)
expected_children = {
'child_node_1': test_child_node_1,
'child_node_2': test_child_node_2,
}
self.assertDictEqual(test_parent_node.get_children(), expected_children)
def test_combined_tests_2(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.add_child(child_node=test_child_node_1)
self.assertEqual(test_parent_node.get_num_children(), 2)
def test_combined_tests_1(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)
self.assertSetEqual(set(test_parent_node_1.get_children_names()), {'child_node_1', '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_1', 'parent_node_2'})
self.assertSetEqual(set(test_child_node_2.get_parents_names()), {'parent_node_1', 'parent_node_2'})
