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