Higher_-_Lower/test.py at main · LiamBolt/Higher_-_Lower · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
# test.py
import unittest
from game import compare_user_guess

data_example = [
    {
        'name': 'Apass',
        'follower_count': 50,
        'description': 'Musician',
        'country': 'Uganda'
    },
    {
        'name': 'Samanya Uganda',
        'follower_count': 45,
        'description': 'Online news platform', 'country': 'Uganda'},
]


class TestHigherLowerGame(unittest.TestCase):
    def test_compare_user_guess_correct_a(self):
        score = 0
        result = compare_user_guess(
            'a', score,
            ['Apass', 50, 'Musician', 'Uganda'],
            ['Samanya Uganda', 45, 'Online news platform', 'Uganda']
        )
        # User guessed 'a', and A has more followers (50 > 45)
        self.assertEqual(result, score + 1)

    def test_compare_user_guess_correct_b(self):
        score = 0
        result = compare_user_guess(
            'b', score,
            ['Apass', 50, 'Musician', 'Uganda'],
            ['Uganda Wildlife Authority', 350,
             'Government agency for wildlife conservation', 'Uganda']
        )
        # User guessed 'b', and B has more followers (350 > 50)
        self.assertEqual(result, score + 1)

    def test_compare_user_guess_incorrect(self):
        score = 0
        result = compare_user_guess(
            'a', score,
            ['Samanya Uganda', 45, 'Online news platform', 'Uganda'],
            ['Uganda Wildlife Authority', 350,
             'Government agency for wildlife conservation', 'Uganda']
        )
        # User guessed 'a', but B has more followers (350 > 45)
        self.assertEqual(result, score)


if __name__ == "__main__":
    unittest.main()