def test_get_destination_success(self): from_node = Node() to_node = Node() test_case = Edge(from_node, to_node) test_result = test_case.get_destination(from_node) self.assertEqual(to_node, test_result, "Wrong destination")