def test_data_print(self):
initial_dataset = {
('Stockholm', 'New York', 'SK22', 2, 'Gate 22, seat 7B',
'Baggage will be automatically transferred from your last leg.')}
boarding_card = TourSortingApi().generate_boarding_cards(initial_dataset)[0]
self.assertEqual(boarding_card.__str__(),
f'Take flight SK22 from Stockholm to New York. Gate 22, seat 7B. Baggage will be '
f'automatically transferred from your last leg.')
def test_unknown_travel_type(self):
initial_dataset = {
('Stockholm', 'New York', 'SK22', 5, 'Gate 22, seat 7B',
'Baggage will be automatically transferred from your last leg.')}
with self.assertRaises(exceptions.UnknownTravelType):
result = TourSortingApi().parse_data(initial_dataset)
print(result[0])
def test_invalid_dataset(self):
initial_dataset = {
('Stockholm', 'New York', 'SK22', 2, 'Gate 22, seat 7B',
'Baggage will be automatically transferred from your last leg.'),
('Barcelona', 'airport bus', 1, '', '')}
with self.assertRaises(exceptions.InvalidBoardingCardData):
TourSortingApi().parse_data(initial_dataset)
def test_broken_chain(self):
initial_dataset = {
('Stockholm', 'New York', 'SK22', 2, 'Gate 22, seat 7B',
'Baggage will be automatically transferred from your last leg.'),
('Barcelona', 'Warsaw', 'airport bus', 1, '', '')}
with self.assertRaises(exceptions.BrokenChainException):
TourSortingApi().parse_data(initial_dataset)
def test_correct_data(self):
initial_dataset = {
('Stockholm', 'New York', 'SK22', 2, 'Gate 22, seat 7B',
'Baggage will be automatically transferred from your last leg.'),
('Barcelona', 'Gerona', 'airport bus', 1, '', ''),
('Madrid', 'Barcelona', '78A', 0, 'seat 45B', ''),
('Gerona', 'Stockholm', 'SK455', 2, 'Gate 45B, seat 3A',
'Baggage drop at ticket counter 344.'),
}
sorted_data = TourSortingApi().parse_data(initial_dataset)
self.assertEqual(sorted_data[0].travel_start, 'Madrid')
self.assertEqual(sorted_data[-1].travel_dest, 'New York')
def main():
# unordered set of sample data. Remember to keep the order of the data!
initial_dataset = {
('Stockholm', 'New York', 'SK22', 2, 'Gate 22, seat 7B',
'Baggage will be automatically transferred from your last leg.'),
('Barcelona', 'Gerona', 'airport bus', 1, '', ''),
('Madrid', 'Barcelona', '78A', 0, 'seat 45B', ''),
('Gerona', 'Stockholm', 'SK455', 2, 'Gate 45B, seat 3A',
'Baggage drop at ticket counter 344.'),
}
for obj in TourSortingApi().parse_data(initial_dataset):
print(obj)
print('You have arrived at your final destination.')
def test_empty_dataset(self):
with self.assertRaises(exceptions.EmptySetException):
TourSortingApi().parse_data(set())
def test_correct_data_one_element(self):
initial_dataset = {
('Stockholm', 'New York', 'SK22', 2, 'Gate 22, seat 7B',
'Baggage will be automatically transferred from your last leg.')}
sorted_data = TourSortingApi().parse_data(initial_dataset)
self.assertEqual(len(sorted_data), 1)
def test_invalid_dataset_type_2(self):
initial_dataset = ['Surely this is not a correct format']
with self.assertRaises(exceptions.InvalidBoardingCardData):
TourSortingApi().parse_data(initial_dataset)
def test_invalid_dataset_type(self):
initial_dataset = {3, 32, 4}
with self.assertRaises(exceptions.InvalidBoardingCardData):
TourSortingApi().parse_data(initial_dataset)