|
1 | 1 | importunittest |
2 | 2 |
|
3 | | -frombook_storeimportcalculate_total |
| 3 | +frombook_storeimporttotal |
4 | 4 |
|
5 | 5 |
|
6 | 6 | # Tests adapted from `problem-specifications//canonical-data.json` @ v1.4.0 |
7 | 7 |
|
8 | 8 | classBookStoreTest(unittest.TestCase): |
9 | 9 | deftest_only_a_single_book(self): |
10 | | -self.assertEqual(calculate_total([1]), 800) |
| 10 | +self.assertEqual(total([1]), 800) |
11 | 11 |
|
12 | 12 | deftest_two_of_the_same_book(self): |
13 | | -self.assertEqual(calculate_total([2, 2]), 1600) |
| 13 | +self.assertEqual(total([2, 2]), 1600) |
14 | 14 |
|
15 | 15 | deftest_empty_basket(self): |
16 | | -self.assertEqual(calculate_total([]), 0) |
| 16 | +self.assertEqual(total([]), 0) |
17 | 17 |
|
18 | 18 | deftest_two_different_books(self): |
19 | | -self.assertEqual(calculate_total([1, 2]), 1520) |
| 19 | +self.assertEqual(total([1, 2]), 1520) |
20 | 20 |
|
21 | 21 | deftest_three_different_books(self): |
22 | | -self.assertEqual(calculate_total([1, 2, 3]), 2160) |
| 22 | +self.assertEqual(total([1, 2, 3]), 2160) |
23 | 23 |
|
24 | 24 | deftest_four_different_books(self): |
25 | | -self.assertEqual(calculate_total([1, 2, 3, 4]), 2560) |
| 25 | +self.assertEqual(total([1, 2, 3, 4]), 2560) |
26 | 26 |
|
27 | 27 | deftest_five_different_books(self): |
28 | | -self.assertEqual(calculate_total([1, 2, 3, 4, 5]), 3000) |
| 28 | +self.assertEqual(total([1, 2, 3, 4, 5]), 3000) |
29 | 29 |
|
30 | 30 | deftest_two_groups_of_4_is_cheaper_than_group_of_5_plus_group_of_3(self): |
31 | | -self.assertEqual(calculate_total([1, 1, 2, 2, 3, 3, 4, 5]), 5120) |
| 31 | +self.assertEqual(total([1, 1, 2, 2, 3, 3, 4, 5]), 5120) |
32 | 32 |
|
33 | 33 | deftest_two_groups_of_4_is_cheaper_than_groups_of_5_and_3(self): |
34 | | -self.assertEqual(calculate_total([1, 1, 2, 3, 4, 4, 5, 5]), 5120) |
| 34 | +self.assertEqual(total([1, 1, 2, 3, 4, 4, 5, 5]), 5120) |
35 | 35 |
|
36 | 36 | deftest_group_of_4_plus_group_of_2_is_cheaper_than_2_groups_of_3(self): |
37 | | -self.assertEqual(calculate_total([1, 1, 2, 2, 3, 4]), 4080) |
| 37 | +self.assertEqual(total([1, 1, 2, 2, 3, 4]), 4080) |
38 | 38 |
|
39 | 39 | deftest_two_each_of_first_4_books_and_1_copy_each_of_rest(self): |
40 | | -self.assertEqual(calculate_total([1, 1, 2, 2, 3, 3, 4, 4, 5]), 5560) |
| 40 | +self.assertEqual(total([1, 1, 2, 2, 3, 3, 4, 4, 5]), 5560) |
41 | 41 |
|
42 | 42 | deftest_two_copies_of_each_book(self): |
43 | | -self.assertEqual(calculate_total([1, 1, 2, 2, 3, 3, 4, 4, 5, 5]), 6000) |
| 43 | +self.assertEqual(total([1, 1, 2, 2, 3, 3, 4, 4, 5, 5]), 6000) |
44 | 44 |
|
45 | 45 | deftest_three_copies_of_first_book_and_2_each_of_remaining(self): |
46 | 46 | self.assertEqual( |
47 | | -calculate_total([1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 1]), 6800) |
| 47 | +total([1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 1]), 6800) |
48 | 48 |
|
49 | 49 | deftest_three_each_of_first_2_books_and_2_each_of_remaining_books(self): |
50 | 50 | self.assertEqual( |
51 | | -calculate_total([1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 1, 2]), 7520) |
| 51 | +total([1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 1, 2]), 7520) |
52 | 52 |
|
53 | 53 | deftest_four_groups_of_4_are_cheaper_than_2_groups_each_of_5_and_3(self): |
54 | 54 | self.assertEqual( |
55 | | -calculate_total([1, 1, 2, 2, 3, 3, 4, 5, 1, 1, 2, 2, 3, 3, 4, 5]), |
| 55 | +total([1, 1, 2, 2, 3, 3, 4, 5, 1, 1, 2, 2, 3, 3, 4, 5]), |
56 | 56 | 10240) |
57 | 57 |
|
58 | 58 |
|
|
0 commit comments