def test_eliminar_vol_calcular_import(self):
p = Viatgers(3)
past = p.get_num_viatgers()
f1 = Vol("CODIGO1", "MILAN", past, 25.0)
f2 = Vol("CODIGO2", "BARCELONA", past, 19.0)
vols = [f1, f2]
v = Flights(past, vols)
v.elimina_desti("BARCELONA")
self.assertEqual(v.calcul_import_total_vols(), 75.0)
def test_eliminar_vol_llista_hotels(self):
p = Viatgers(3)
past = p.get_num_viatgers()
f1 = Vol("CODIGO1", "MILAN", past, 25.0)
f2 = Vol("CODIGO2", "BARCELONA", past, 19.0)
vols = [f1, f2]
v = Flights(past, vols)
v.elimina_desti("BARCELONA")
test_llista_vols = [f1]
self.assertEqual(v.get_llista_vols(), test_llista_vols)
