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)