def test_get_edge_returns_cost_when_direct_flights_available_case2(
filled_weighted_graph):
"""get_edges() returns the total cost when there are direct flights available between all the cities in the itinerary
"""
itinerary = ['Metroville', 'Pandora']
actual = get_edges(filled_weighted_graph, itinerary)
assert actual == 'TRRRRRRRRUUUUUUUUEEEEEEEE!!!!!!!!!! btw, total cost is 82'
def test_get_edge_returns_false_when_no_direct_flights_case2(
filled_weighted_graph):
"""get_edges() returns false when there are no direct flights between two of the cities in the itinerary
"""
itinerary = ['Narnia', 'Arendelle', 'Naboo']
actual = get_edges(filled_weighted_graph, itinerary)
assert not actual
def test_get_edge_returns_cost_when_direct_flights_available_case1(
filled_weighted_graph):
"""get_edges() returns the total cost when there are direct flights available between all the cities in the itinerary
"""
itinerary = ['Arendelle', 'Monstropolis', 'Naboo']
actual = get_edges(filled_weighted_graph, itinerary)
assert actual == 'TRRRRRRRRUUUUUUUUEEEEEEEE!!!!!!!!!! btw, total cost is 115'
def test_returns_false_when_given_cities_that_do_not_exist(
filled_weighted_graph):
"""get_edges() returns false when given an itinerary including cities that are airline does not service
"""
itinerary = ['Seattle', 'New York City']
actual = get_edges(filled_weighted_graph, itinerary)
assert not actual