def test_get_all_routes_simple_2_sets_of_2_route(self):
data = [
{
"start": "Buenos Aires",
"end": "New York",
"journey_time": 6
},
{
"start": "New York",
"end": "Liverpool",
"journey_time": 4
},
{
"start": "Buenos Aires",
"end": "Casablanca",
"journey_time": 5
},
{
"start": "Casablanca",
"end": "Liverpool",
"journey_time": 3
},
]
rm = RouteManager()
rm.set_routes(data)
a = rm.get_all_routes("Buenos Aires", "Liverpool")
self.assertListEqual([
["Buenos Aires", "New York", "Liverpool"],
["Buenos Aires", "Casablanca", "Liverpool"],
], a)
def test_get_all_routes_simple_1_set_1_route(self):
data = [
{
"start": "Buenos Aires",
"end": "New York",
"journey_time": 6
},
]
rm = RouteManager()
rm.set_routes(data)
a = rm.get_all_routes("Buenos Aires", "New York")
self.assertListEqual([["Buenos Aires", "New York"]], a)
def test_get_all_routes_round_trip(self):
data = [
{
"start": "Liverpool",
"end": "Casablanca",
"journey_time": 3
},
{
"start": "Casablanca",
"end": "Liverpool",
"journey_time": 3
},
]
rm = RouteManager()
rm.set_routes(data)
a = rm.get_all_routes("Liverpool", "Liverpool")
self.assertListEqual([["Liverpool", "Casablanca", "Liverpool"]], a)
def test_get_all_routes_Invalid_port_names_1(self):
"""No ports specified"""
rm = RouteManager()
rm.set_routes([])
with self.assertRaises(InvalidPortName):
rm.get_all_routes("Buenos Aires", "Casablanca")