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")
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")