def test_get_num_parents(self):
test_child_node = UnorderedNodeBase(node_name='child_node')
test_parent_node_1 = UnorderedNodeBase(node_name='parent_node_1')
test_parent_node_2 = UnorderedNodeBase(node_name='parent_node_2')
test_child_node.add_parent(parent_node=test_parent_node_1)
test_child_node.add_parent(parent_node=test_parent_node_2)
self.assertEqual(test_child_node.get_num_parents(), 2)
def test_get_parents_nodes(self):
test_child_node = UnorderedNodeBase(node_name='child_node')
test_parent_node_1 = UnorderedNodeBase(node_name='parent_node_1')
test_parent_node_2 = UnorderedNodeBase(node_name='parent_node_2')
test_child_node.add_parent(parent_node=test_parent_node_1)
test_child_node.add_parent(parent_node=test_parent_node_2)
self.assertSetEqual(set(test_child_node.get_parents_nodes()), {test_parent_node_1, test_parent_node_2})
def test_has_parent(self):
test_child_node = UnorderedNodeBase(node_name='child_node')
test_parent_node = UnorderedNodeBase(node_name='parent_node')
test_child_node.add_parent(parent_node=test_parent_node)
self.assertTrue(test_child_node.has_parent(parent_name='parent_node'))
self.assertFalse(test_child_node.has_parent(parent_name='parent_node_0'))
def test_get_parent(self):
test_child_node = UnorderedNodeBase(node_name='child_node')
test_parent_node = UnorderedNodeBase(node_name='parent_node')
test_child_node.add_parent(parent_node=test_parent_node)
self.assertEqual(test_child_node.get_parent(parent_name='parent_node'), test_parent_node)
self.assertEqual(test_child_node.get_parent(parent_name='parent_node_0'), None)
def test_get_parents(self):
test_child_node = UnorderedNodeBase(node_name='child_node')
test_parent_node_1 = UnorderedNodeBase(node_name='parent_node_1')
test_parent_node_2 = UnorderedNodeBase(node_name='parent_node_2')
test_child_node.add_parent(parent_node=test_parent_node_1)
test_child_node.add_parent(parent_node=test_parent_node_2)
expected_parents = {
'parent_node_1': test_parent_node_1,
'parent_node_2': test_parent_node_2,
}
self.assertDictEqual(test_child_node.get_parents(), expected_parents)
