exercism-solution/python/perfect-numbers/perfect_numbers_test.py at master · fruit-in/exercism-solution · 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
55
56
57
58
59
60
61
62
import unittest

from perfect_numbers import classify

# Tests adapted from `problem-specifications//canonical-data.json`


class PerfectNumbersTest(unittest.TestCase):
    def test_smallest_perfect_number_is_classified_correctly(self):
        self.assertIs(classify(6), "perfect")

    def test_medium_perfect_number_is_classified_correctly(self):
        self.assertIs(classify(28), "perfect")

    def test_large_perfect_number_is_classified_correctly(self):
        self.assertIs(classify(33550336), "perfect")


class AbundantNumbersTest(unittest.TestCase):
    def test_smallest_abundant_number_is_classified_correctly(self):
        self.assertIs(classify(12), "abundant")

    def test_medium_abundant_number_is_classified_correctly(self):
        self.assertIs(classify(30), "abundant")

    def test_large_abundant_number_is_classified_correctly(self):
        self.assertIs(classify(33550335), "abundant")


class DeficientNumbersTest(unittest.TestCase):
    def test_smallest_prime_deficient_number_is_classified_correctly(self):
        self.assertIs(classify(2), "deficient")

    def test_smallest_non_prime_deficient_number_is_classified_correctly(self):
        self.assertIs(classify(4), "deficient")

    def test_medium_deficient_number_is_classified_correctly(self):
        self.assertIs(classify(32), "deficient")

    def test_large_deficient_number_is_classified_correctly(self):
        self.assertIs(classify(33550337), "deficient")

    def test_edge_case_no_factors_other_than_itself_is_classified_correctly(self):
        self.assertIs(classify(1), "deficient")


class InvalidInputsTest(unittest.TestCase):
    def test_zero_is_rejected_as_it_is_not_a_positive_integer(self):
        with self.assertRaisesWithMessage(ValueError):
            classify(0)

    def test_negative_integer_is_rejected_as_it_is_not_a_positive_integer(self):
        with self.assertRaisesWithMessage(ValueError):
            classify(-1)

    # Utility functions
    def assertRaisesWithMessage(self, exception):
        return self.assertRaisesRegex(exception, r".+")


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