|
1 | 1 | import unittest |
2 | 2 |
|
3 | | -from high_scores import HighScores |
| 3 | +from high_scores import latest, personal_best, personal_top_three |
4 | 4 |
|
5 | 5 |
|
6 | 6 | # Tests adapted from `problem-specifications//canonical-data.json` @ v4.0.0 |
7 | 7 |
|
8 | 8 |
|
9 | 9 | class HighScoreTest(unittest.TestCase): |
10 | | - def test_list_of_scores(self): |
11 | | - scores = [30, 50, 20, 70] |
12 | | - expected = [30, 50, 20, 70] |
13 | | - self.assertEqual(HighScores(scores).scores, expected) |
14 | 10 |
|
15 | 11 | def test_latest_score(self): |
16 | 12 | scores = [100, 0, 90, 30] |
17 | 13 | expected = 30 |
18 | | - self.assertEqual(HighScores(scores).latest(), expected) |
| 14 | + self.assertEqual(latest(scores), expected) |
19 | 15 |
|
20 | 16 | def test_personal_best(self): |
21 | 17 | scores = [40, 100, 70] |
22 | 18 | expected = 100 |
23 | | - self.assertEqual(HighScores(scores).personal_best(), expected) |
| 19 | + self.assertEqual(personal_best(scores), expected) |
24 | 20 |
|
25 | 21 | def test_personal_top_three_from_a_long_list(self): |
26 | 22 | scores = [10, 30, 90, 30, 100, 20, 10, 0, 30, 40, 40, 70, 70] |
27 | 23 | expected = [100, 90, 70] |
28 | | - self.assertEqual(HighScores(scores).personal_top_three(), expected) |
| 24 | + self.assertEqual(personal_top_three(scores), expected) |
29 | 25 |
|
30 | 26 | def test_personal_top_three_highest_to_lowest(self): |
31 | 27 | scores = [20, 10, 30] |
32 | 28 | expected = [30, 20, 10] |
33 | | - self.assertEqual(HighScores(scores).personal_top_three(), expected) |
| 29 | + self.assertEqual(personal_top_three(scores), expected) |
34 | 30 |
|
35 | 31 | def test_personal_top_three_when_there_is_a_tie(self): |
36 | 32 | scores = [40, 20, 40, 30] |
37 | 33 | expected = [40, 40, 30] |
38 | | - self.assertEqual(HighScores(scores).personal_top_three(), expected) |
| 34 | + self.assertEqual(personal_top_three(scores), expected) |
39 | 35 |
|
40 | 36 | def test_personal_top_three_when_there_are_less_than_3(self): |
41 | 37 | scores = [30, 70] |
42 | 38 | expected = [70, 30] |
43 | | - self.assertEqual(HighScores(scores).personal_top_three(), expected) |
| 39 | + self.assertEqual(personal_top_three(scores), expected) |
44 | 40 |
|
45 | 41 | def test_personal_top_three_when_there_is_only_one(self): |
46 | 42 | scores = [40] |
47 | 43 | expected = [40] |
48 | | - self.assertEqual(HighScores(scores).personal_top_three(), expected) |
| 44 | + self.assertEqual(personal_top_three(scores), expected) |
49 | 45 |
|
50 | 46 |
|
51 | 47 | if __name__ == "__main__": |
|
0 commit comments