python/test_jmbg.py at main · jmbg-labs/python · 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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
import datetime
import unittest

from jmbg import Jmbg, parse, valid
from jmbg_error import JmbgError


class TestJmbg(unittest.TestCase):
    def test_valid_jmbg_can_be_parsed(self):
        jmbg = Jmbg.parse('0710003730015')
        self.assertIsInstance(jmbg, Jmbg)

    def test_valid_jmbg_returns_true(self):
        self.assertTrue(valid('0710003730015'))

    def test_invalid_jmbg_returns_false(self):
        self.assertFalse(valid('1234567890123'))

    def test_invalid_length_throws_exception(self):
        with self.assertRaises(JmbgError) as context:
            Jmbg('123456789')
        self.assertIn('13 digits', str(context.exception))

    def test_non_numeric_jmbg_throws_exception(self):
        with self.assertRaises(JmbgError) as context:
            Jmbg('01019907100ab')
        self.assertIn('numeric', str(context.exception))

    def test_invalid_date_throws_exception(self):
        with self.assertRaises(JmbgError) as context:
            Jmbg('3201990710009')
        self.assertRegex(str(context.exception), r'Date .* is not valid')

    def test_invalid_region_throws_exception(self):
        with self.assertRaises(JmbgError) as context:
            Jmbg('0710003660015')
        self.assertEqual("Region '66' is not valid for JMBG", str(context.exception))

    def test_invalid_checksum_throws_exception(self):
        with self.assertRaises(JmbgError) as context:
            Jmbg('0710003730025')
        self.assertEqual('Checksum is not valid', str(context.exception))

    def test_is_male_returns_true(self):
        jmbg = Jmbg('0710003730015')
        self.assertTrue(jmbg.is_male())

    def test_is_male_returns_false(self):
        jmbg = Jmbg('0710003735017')
        self.assertFalse(jmbg.is_male())

    def test_is_female_returns_true(self):
        jmbg = Jmbg('0710003735017')
        self.assertTrue(jmbg.is_female())

    def test_is_female_returns_false(self):
        jmbg = Jmbg('0710003730015')
        self.assertFalse(jmbg.is_female())

    def test_get_age(self):
        jmbg = Jmbg('0710003730015')
        birth_date = datetime.date(2003, 10, 7)
        today = datetime.date.today()
        expected_age = today.year - birth_date.year
        if (today.month, today.day) < (birth_date.month, birth_date.day):
            expected_age -= 1
        self.assertEqual(expected_age, jmbg.get_age())

    def test_get_date(self):
        jmbg = Jmbg('0710003730015')
        date = jmbg.get_date()
        self.assertIsInstance(date, datetime.date)
        self.assertEqual('2003-10-07', date.strftime('%Y-%m-%d'))

    def test_format(self):
        jmbg = Jmbg('0710003730015')
        self.assertEqual('0710003730015', jmbg.format())

    def test_to_string(self):
        jmbg = Jmbg('0710003730015')
        self.assertEqual('0710003730015', str(jmbg))

    def test_attributes(self):
        jmbg = Jmbg('2902992710005')

        self.assertEqual('2902992710005', jmbg.original)
        self.assertEqual(29, jmbg.day)
        self.assertEqual('29', jmbg.day_original)
        self.assertEqual(2, jmbg.month)
        self.assertEqual('02', jmbg.month_original)
        self.assertEqual(1992, jmbg.year)
        self.assertEqual('992', jmbg.year_original)
        self.assertEqual(71, jmbg.region)
        self.assertEqual('71', jmbg.region_original)
        self.assertEqual('Belgrade', jmbg.region_text)
        self.assertEqual('Serbia', jmbg.country)
        self.assertEqual(0, jmbg.unique)
        self.assertEqual('000', jmbg.unique_original)
        self.assertEqual(5, jmbg.checksum)

    def test_year_calculation_for_2000s(self):
        jmbg = Jmbg('0101000710009')
        self.assertEqual(2000, jmbg.year)

    def test_year_calculation_for_1900s(self):
        jmbg = Jmbg('1705978730032')
        self.assertEqual(1978, jmbg.year)

    def test_different_regions(self):
        # Belgrade
        jmbg1 = Jmbg('2902992710005')
        self.assertEqual('Belgrade', jmbg1.region_text)
        self.assertEqual('Serbia', jmbg1.country)

        # Novi Sad
        jmbg2 = Jmbg('1505995800002')
        self.assertEqual('Novi Sad', jmbg2.region_text)
        self.assertEqual('Serbia', jmbg2.country)

    def test_boundary_unique_numbers(self):
        # Male - unique 0
        jmbg1 = Jmbg('1505995800002')
        self.assertTrue(jmbg1.is_male())
        self.assertEqual(0, jmbg1.unique)

        # Male - unique 499
        jmbg2 = Jmbg('1505995804997')
        self.assertTrue(jmbg2.is_male())
        self.assertEqual(499, jmbg2.unique)

        # Female - unique 500
        jmbg3 = Jmbg('1505995805004')
        self.assertTrue(jmbg3.is_female())
        self.assertEqual(500, jmbg3.unique)

        # Female - unique 999
        jmbg4 = Jmbg('1505995809999')
        self.assertTrue(jmbg4.is_female())
        self.assertEqual(999, jmbg4.unique)

    def test_leap_year_date(self):
        jmbg = Jmbg('2902992710005')
        self.assertEqual(29, jmbg.day)
        self.assertEqual(2, jmbg.month)
        self.assertEqual(1992, jmbg.year)

    def test_invalid_leap_year_date_throws_exception(self):
        with self.assertRaises(JmbgError) as context:
            Jmbg('2902979758318')
        self.assertRegex(str(context.exception), r'Date .* is not valid')


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