PostCodes/tests.py at master · garyb1/PostCodes · 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
#!/usr/bin/python3

"""
Testing was done on a random list of valid
postcodes created from http://www.ukpostcode.co.uk/random.htm

Most important functions:
    - is_postcode_valid
    - get_postcode_date

"""


import unittest
from postcode.postcode import *


class TestLibraryMethods(unittest.TestCase):

    def test_postcode_validity(self):
        self.assertTrue(is_postcode_valid("NR9 4QJ"))
        self.assertTrue(is_postcode_valid("CF83 1UQ"))
        self.assertTrue(is_postcode_valid("N1 8AL"))
        self.assertTrue(is_postcode_valid("PL4 8LL"))
        self.assertTrue(is_postcode_valid("CR0 8QD"))
        self.assertTrue(is_postcode_valid("RG27 9HW"))
        self.assertTrue(is_postcode_valid("HX3 0ST"))
        self.assertTrue(is_postcode_valid("EH12 7RJ"))

        self.assertFalse(is_postcode_valid("N29 422SJ"))
        self.assertFalse(is_postcode_valid("CFDS 1UQ"))
        self.assertFalse(is_postcode_valid("N13 8XL1"))
        self.assertFalse(is_postcode_valid("PX4 8L"))
        self.assertFalse(is_postcode_valid("CR2 20D"))
        self.assertFalse(is_postcode_valid("RGS7 9HW"))
        self.assertFalse(is_postcode_valid("HXSW 0ST"))
        self.assertFalse(is_postcode_valid("EH1 78J"))

    def test_nearby_function(self):
        self.assertTrue(get_nearest_postcodes("NR9 4QJ"))
        self.assertTrue(get_nearest_postcodes("CR0 8QD"))
        assert not get_nearest_postcodes("CFDS 1UQ")

    def test_postcode_data(self):
        # 202 response code means postcode is valid
        self.assertEqual(get_postcode_data("NR9 4QJ")["status"], 200)
        # 404 response code means postcode not found
        self.assertEqual(get_postcode_data("CFDS 4QJ")["status"], 404)

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