def test_viaje_eliminar_destinos(self) -> None: lVuelos = Flights(num_pasajeros=self.num_pasajeros) for destino in self.lDestinacion_vuelos: lVuelos.add_destino(destino, 0) for destinoEliminar in self.lDestinos_eliminar: lVuelos.eliminar_destino(destinoEliminar) self.assertEqual(lVuelos.get_destinos(), ["Suecia"])
def test_precio_viaje_esperado_eliminar_destinos( self) -> None: # volveremos vuelos = Flights(num_pasajeros=self.num_pasajeros) for destino, precio in zip(self.lDestinacion_vuelos, self.lPrecios): vuelos.add_destino(destino, precio) # se elimian 2 destinos vuelos.eliminar_destino("Dinamarcia") vuelos.eliminar_destino("Suecia") # se descuetnan 2 destinos en el valor esperado lPrecios = [30, 0, 0] importeEsperado = sum(lPrecios) * self.num_pasajeros assert vuelos.get_total() == importeEsperado print("Test_precio_viaje_esperado_eliminar_destinos")