@@ -14,10 +14,15 @@ def test_get_list_of_wagons(self):
14
14
input_data = [(1 ,5 ,2 ,7 ,4 ), (1 ,5 ), (1 ,), (1 ,9 ,3 ), (1 ,10 ,6 ,3 ,9 ,8 ,4 ,14 ,24 ,7 )]
15
15
output_data = [[1 ,5 ,2 ,7 ,4 ], [1 ,5 ], [1 ], [1 ,9 ,3 ], [1 ,10 ,6 ,3 ,9 ,8 ,4 ,14 ,24 ,7 ]]
16
16
17
- for variant , (input_data , output_data ) in enumerate (zip (input_data , output_data ), start = 1 ):
18
- with self .subTest (f'variation #{ variant } ' , input_data = input_data , output_data = output_data ):
19
- error_msg = f'Expected: { output_data } but got a different wagon list instead.'
20
- self .assertEqual (get_list_of_wagons (* input_data ), output_data , msg = error_msg )
17
+ for variant , (input_data , expected ) in enumerate (zip (input_data , output_data ), start = 1 ):
18
+ with self .subTest (f'variation #{ variant } ' , input_data = input_data , expected = expected ):
19
+
20
+ actual_result = get_list_of_wagons (* input_data )
21
+ error_msg = (f'Called get_list_of_wagons{ input_data } . '
22
+ f'The function returned { actual_result } , but the '
23
+ f'tests expected: { expected } as the wagon list instead.' )
24
+
25
+ self .assertEqual (actual_result , expected , msg = error_msg )
21
26
22
27
@pytest .mark .task (taskno = 2 )
23
28
def test_fix_list_of_wagons (self ):
@@ -31,10 +36,15 @@ def test_fix_list_of_wagons(self):
31
36
[1 , 8 , 6 , 15 , 4 , 2 ],
32
37
[1 , 8 , 6 , 4 , 5 , 9 , 21 , 2 , 13 , 25 , 7 , 19 , 10 , 3 , 14 ]
33
38
]
34
- for variant , (input_data , output_data ) in enumerate (zip (input_data , output_data ), start = 1 ):
35
- with self .subTest (f'variation #{ variant } ' , input_data = input_data , output_data = output_data ):
36
- error_msg = f'Expected: { output_data } but got a different wagon list instead.'
37
- self .assertEqual (fix_list_of_wagons (input_data [0 ], input_data [1 ]), output_data , msg = error_msg )
39
+ for variant , (input_data , expected ) in enumerate (zip (input_data , output_data ), start = 1 ):
40
+ with self .subTest (f'variation #{ variant } ' , input_data = input_data , expected = expected ):
41
+
42
+ actual_result = fix_list_of_wagons (input_data [0 ], input_data [1 ])
43
+ error_msg = (f'Called fix_list_of_wagons({ input_data [0 ]} , { input_data [1 ]} ). '
44
+ f'The function returned { actual_result } , but the '
45
+ f'tests expected: { expected } as the wagon list instead.' )
46
+
47
+ self .assertEqual (actual_result , expected , msg = error_msg )
38
48
39
49
@pytest .mark .task (taskno = 3 )
40
50
def test_add_missing_stops (self ):
@@ -48,10 +58,15 @@ def test_add_missing_stops(self):
48
58
{'from' : 'New York' , 'to' : 'Philadelphia' , 'stops' : []},
49
59
{'from' : 'Gothenburg' , 'to' : 'Copenhagen' , 'stops' : ['Kungsbacka' , 'Varberg' , 'Halmstad' , 'Angelholm' , 'Lund' , 'Malmo' ]}
50
60
]
51
- for variant , (input_data , output_data ) in enumerate (zip (input_data , output_data ), start = 1 ):
52
- with self .subTest (f'variation #{ variant } ' , input_data = input_data , output_data = output_data ):
53
- error_msg = f'Expected: { output_data } but got a different set of stops instead.'
54
- self .assertEqual (add_missing_stops (input_data [0 ], ** input_data [1 ]), output_data , msg = error_msg )
61
+ for variant , (input_data , expected ) in enumerate (zip (input_data , output_data ), start = 1 ):
62
+ with self .subTest (f'variation #{ variant } ' , input_data = input_data , expected = expected ):
63
+
64
+ actual_result = add_missing_stops (input_data [0 ], ** input_data [1 ])
65
+ error_msg = (f'Called add_missing_stops({ input_data [0 ]} , { input_data [1 ]} ). '
66
+ f'The function returned { actual_result } , but the '
67
+ f'tests expected: { expected } as the set of stops.' )
68
+
69
+ self .assertEqual (actual_result , expected , msg = error_msg )
55
70
56
71
@pytest .mark .task (taskno = 4 )
57
72
def test_extend_route_information (self ):
@@ -63,10 +78,15 @@ def test_extend_route_information(self):
63
78
{'from' : 'Gothenburg' , 'to' : 'Copenhagen' , 'precipitation' : '1' , 'timeOfArrival' : '21:20' , 'temperature' : '-6' }
64
79
]
65
80
66
- for variant , (input_data , output_data ) in enumerate (zip (input_data , output_data ), start = 1 ):
67
- with self .subTest (f'variation #{ variant } ' , input_data = input_data , output_data = output_data ):
68
- error_msg = f'Expected: { output_data } but got a different route dictionary instead.'
69
- self .assertEqual (extend_route_information (input_data [0 ], input_data [1 ]), output_data , msg = error_msg )
81
+ for variant , (input_data , expected ) in enumerate (zip (input_data , output_data ), start = 1 ):
82
+ with self .subTest (f'variation #{ variant } ' , input_data = input_data , expected = expected ):
83
+
84
+ actual_result = extend_route_information (input_data [0 ], input_data [1 ])
85
+ error_msg = (f'Called extend_route_information({ input_data [0 ]} , { input_data [1 ]} ). '
86
+ f'The function returned { actual_result } , but the '
87
+ f'tests expected: { expected } as the route dictionary.' )
88
+
89
+ self .assertEqual (actual_result , expected , msg = error_msg )
70
90
71
91
@pytest .mark .task (taskno = 5 )
72
92
def test_fix_wagon_depot (self ):
@@ -84,7 +104,12 @@ def test_fix_wagon_depot(self):
84
104
[[(3 , 'purple' ), (20 , 'black' ), (19 , 'white' )], [(11 , 'purple' ), (16 , 'black' ), (17 , 'white' )], [(15 , 'purple' ), (12 , 'black' ), (18 , 'white' )]]
85
105
)
86
106
87
- for variant , (input_data , output_data ) in enumerate (zip (input_data , output_data ), start = 1 ):
88
- with self .subTest (f'variation #{ variant } ' , input_data = input_data , output_data = output_data ):
89
- error_msg = f'Expected: { output_data } but got a different wagon depot list instead.'
90
- self .assertEqual (fix_wagon_depot (input_data ), output_data , msg = error_msg )
107
+ for variant , (input_data , expected ) in enumerate (zip (input_data , output_data ), start = 1 ):
108
+ with self .subTest (f'variation #{ variant } ' , input_data = input_data , expected = expected ):
109
+
110
+ actual_result = fix_wagon_depot (input_data )
111
+ error_msg = (f'Called fix_wagon_depot({ input_data } ). '
112
+ f'The function returned { actual_result } , but the '
113
+ f'tests expected: { expected } as the wagon depot list.' )
114
+
115
+ self .assertEqual (actual_result , expected , msg = error_msg )
0 commit comments