|
4 | 4 |
|
5 | 5 |
|
6 | 6 | class MaxRunTimeTestCase(unittest.TestCase): |
7 | | - |
8 | | - @parameterized.expand([ |
9 | | - ([2,3,3,4], 3, 4), |
10 | | - ([1,1,4,5], 2, 5), |
11 | | - ([2,2,2,2], 1, 8), |
12 | | - ([7,2,5,10,8], 2, 16), |
13 | | - ([1,2,3,4,5], 2, 7), |
14 | | - ([3,4,3,4,5,5,8,2], 4, 8), |
15 | | - ([5,2,4], 2, 5), |
16 | | - ([1,6,2,6,8], 5, 1) |
17 | | - ]) |
| 7 | + @parameterized.expand( |
| 8 | + [ |
| 9 | + ([2, 3, 3, 4], 3, 4), |
| 10 | + ([1, 1, 4, 5], 2, 5), |
| 11 | + ([2, 2, 2, 2], 1, 8), |
| 12 | + ([7, 2, 5, 10, 8], 2, 16), |
| 13 | + ([1, 2, 3, 4, 5], 2, 7), |
| 14 | + ([3, 4, 3, 4, 5, 5, 8, 2], 4, 8), |
| 15 | + ([5, 2, 4], 2, 5), |
| 16 | + ([1, 6, 2, 6, 8], 5, 1), |
| 17 | + ] |
| 18 | + ) |
18 | 19 | def test_max_runtime_1(self, batteries, n, expected): |
19 | 20 | actual = max_runtime(batteries, n) |
20 | 21 | self.assertEqual(expected, actual) |
21 | 22 |
|
22 | | - @parameterized.expand([ |
23 | | - ([2,3,3,4], 3, 4), |
24 | | - ([1,1,4,5], 2, 5), |
25 | | - ([2,2,2,2], 1, 8), |
26 | | - ([7,2,5,10,8], 2, 16), |
27 | | - ([1,2,3,4,5], 2, 7), |
28 | | - ([3,4,3,4,5,5,8,2], 4, 8), |
29 | | - ([5,2,4], 2, 5), |
30 | | - ([1,6,2,6,8], 5, 1) |
31 | | - ]) |
| 23 | + @parameterized.expand( |
| 24 | + [ |
| 25 | + ([2, 3, 3, 4], 3, 4), |
| 26 | + ([1, 1, 4, 5], 2, 5), |
| 27 | + ([2, 2, 2, 2], 1, 8), |
| 28 | + ([7, 2, 5, 10, 8], 2, 16), |
| 29 | + ([1, 2, 3, 4, 5], 2, 7), |
| 30 | + ([3, 4, 3, 4, 5, 5, 8, 2], 4, 8), |
| 31 | + ([5, 2, 4], 2, 5), |
| 32 | + ([1, 6, 2, 6, 8], 5, 1), |
| 33 | + ] |
| 34 | + ) |
32 | 35 | def test_max_runtime_2(self, batteries, n, expected): |
33 | 36 | actual = max_run_time_2(batteries, n) |
34 | 37 | self.assertEqual(expected, actual) |
35 | 38 |
|
36 | 39 |
|
37 | | -if __name__ == '__main__': |
| 40 | +if __name__ == "__main__": |
38 | 41 | unittest.main() |
0 commit comments