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)