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
|
# -*- coding: utf-8 -*-
import pytest
import iso3166
from iso3166 import countries
def check_lookup(alpha2, matching_keys, missing_keys):
for k in matching_keys:
assert countries[k].alpha2 == alpha2
assert countries.get(k).alpha2 == alpha2
assert k in countries
for k in missing_keys:
with pytest.raises(KeyError):
countries.get(k)
with pytest.raises(KeyError):
countries[k]
assert countries.get(k, None) is None
def test_length():
assert len(countries) == len(iso3166._records)
def test_empty_string():
check_lookup("US", ["us", "US"], [""])
def test_none():
check_lookup("US", ["us", "US"], [None])
def test_alpha2():
check_lookup("US", ["us", "US"], ["zz"])
def test_alpha3():
check_lookup("US", ["usa", "USA"], ["zzz"])
def test_name():
check_lookup("US", ["united states of america", "United STates of america"], ["zzzzz"])
def test_numeric():
check_lookup("US", [840, "840"], [111, "111"])
def test_alt_name():
check_lookup("TW", ["taiwan", "Taiwan, province of china"], ["zzzzz"])
check_lookup("PS", ["palestine", "palestine, state of"], ["zzzz"])
def test_data():
assert len(list(countries)) > 0
for country in countries:
assert len(country.alpha2) == 2
assert country.alpha2.upper() == country.alpha2
assert len(country.alpha3) == 3
assert country.alpha3.upper() == country.alpha3
assert len(country.numeric) == 3
assert country.numeric == ("%03d" % int(country.numeric))
assert len(country.name) > 3
assert len(country.apolitical_name) > 3
|
