def test_group_only_if_all_parameters_match(self):
# Given
first_train_1 = TrainRoute('', '', '', '', '0', '', '')
first_train_2 = TrainRoute('', '', '', '', '0', '', '')
first_train_3 = TrainRoute('', '', '', '', '0', '', '')
second_train_1 = TrainRoute('', '', '', '', '1', '', '')
second_train_2 = TrainRoute('', '', '', '', '1', '', '')
second_train_3 = TrainRoute('', '', '', '', '1', '', '')
route_1 = Route(first_train_1, second_train_1, '4')
route_2 = Route(first_train_2, second_train_2, '5')
route_3 = Route(first_train_3, second_train_3, '6')
routes = [route_1, route_2, route_3]
# When
grouped = Route.group_routes(routes)
# Then
self.assertEqual(3, len(grouped))
self.assertEqual(['0', '1', '4'], [
grouped[0].first_train.number_train,
grouped[0].second_train.number_train, grouped[0].total_time
])
self.assertEqual(['0', '1', '5'], [
grouped[1].first_train.number_train,
grouped[1].second_train.number_train, grouped[1].total_time
])
self.assertEqual(['0', '1', '6'], [
grouped[2].first_train.number_train,
grouped[2].second_train.number_train, grouped[2].total_time
])
def test_group_all_routes_same(self):
first_train = TrainRoute('', '', '', '', '3310', '', '')
second_train = TrainRoute('', '', '', '', '3310', '', '')
route_1 = Route(first_train, second_train, '12')
route_2 = Route(first_train, second_train, '12')
route_3 = Route(first_train, second_train, '12')
routes = [route_1, route_2, route_3]
grouped = Route.group_routes(routes)
self.assertEqual(1, len(grouped))
self.assertEqual(['3310', '3310', '12'], [
grouped[0].first_train.number_train,
grouped[0].second_train.number_train, grouped[0].total_time
])
def test_group_some_routes_same(self):
# Given
first_train_1 = TrainRoute('', '', '', '', '0', '', '')
first_train_2 = TrainRoute('', '', '', '', '1', '', '')
first_train_3 = TrainRoute('', '', '', '', '2', '', '')
first_train_4 = TrainRoute('', '', '', '', '1', '', '')
first_train_5 = TrainRoute('', '', '', '', '0', '', '')
second_train_1 = TrainRoute('', '', '', '', '0', '', '')
second_train_2 = TrainRoute('', '', '', '', '1', '', '')
second_train_3 = TrainRoute('', '', '', '', '2', '', '')
second_train_4 = TrainRoute('', '', '', '', '1', '', '')
second_train_5 = TrainRoute('', '', '', '', '0', '', '')
route_1 = Route(first_train_1, second_train_1, '1')
route_2 = Route(first_train_2, second_train_2, '2')
route_3 = Route(first_train_3, second_train_3, '3')
route_4 = Route(first_train_4, second_train_4, '2')
route_5 = Route(first_train_5, second_train_5, '1')
routes = [route_1, route_2, route_3, route_4, route_5]
# When
grouped = Route.group_routes(routes)
# Then
self.assertEqual(3, len(grouped))
self.assertEqual(['0', '0', '1'], [
grouped[0].first_train.number_train,
grouped[0].second_train.number_train, grouped[0].total_time
])
self.assertEqual(['1', '1', '2'], [
grouped[1].first_train.number_train,
grouped[1].second_train.number_train, grouped[1].total_time
])
self.assertEqual(['2', '2', '3'], [
grouped[2].first_train.number_train,
grouped[2].second_train.number_train, grouped[2].total_time
])