def test_9(self): aux1 = Vuelos() aux2 = Vuelos(destino='NYC') aux_vuelo = [aux1, aux2] x = Viajes(lista_pasajeros=['p1', 'p2', 'p3'], vuelos=aux_vuelo) x.eliminar_destino('BCN') assert ['NYC'] == x.lista_destinos
def test_10(self): aux1 = Vuelos(destino='BCN') aux2 = Vuelos(destino='ITA') aux_vuelo = [aux1, aux2] x = Viajes(lista_pasajeros=['p1', 'p2', 'p3'], vuelos=aux_vuelo) x.eliminar_destino('BCN') aux_vuelo.remove(aux1) assert aux_vuelo == x.vuelos
def test_11(self): aux1 = Vuelos(destino='BCN') aux2 = Vuelos(destino='ITA') aux_vuelo = [aux1, aux2] x = Viajes(lista_pasajeros=['p1', 'p2', 'p3'], vuelos=aux_vuelo) x.eliminar_destino('BCN') precio_total = 50 * 3 assert precio_total == x.calcular_precio()