|
1 | 1 | import pytest
|
2 | 2 |
|
3 |
| -from fifacodes import Counties |
| 3 | +from fifacodes import Countries |
4 | 4 |
|
5 | 5 |
|
6 |
| -class TestCounties: |
| 6 | +class TestCountries: |
7 | 7 | @pytest.fixture(scope="class")
|
8 |
| - def counties(self) -> Counties: |
9 |
| - return Counties() |
| 8 | + def countries(self) -> Countries: |
| 9 | + return Countries() |
10 | 10 |
|
11 |
| - def test_read_data(self, counties: Counties) -> None: |
12 |
| - assert len(counties._default_data) == 211 |
13 |
| - assert len(counties._data) == 423 |
| 11 | + def test_read_data(self, countries: Countries) -> None: |
| 12 | + assert len(countries._default_data) == 211 |
| 13 | + assert len(countries._data) == 423 |
14 | 14 |
|
15 |
| - def test_init(self, counties: Counties) -> None: |
16 |
| - assert len(counties) == 211 |
| 15 | + def test_init(self, countries: Countries) -> None: |
| 16 | + assert len(countries) == 211 |
17 | 17 |
|
18 |
| - def test_get_key(self, counties: Counties) -> None: |
19 |
| - country = counties.get("ENG") |
| 18 | + def test_get_key(self, countries: Countries) -> None: |
| 19 | + country = countries.get("ENG") |
20 | 20 | if country is not None:
|
21 | 21 | assert country.code == "ENG"
|
22 | 22 | assert country.name == "England"
|
23 | 23 |
|
24 |
| - def test_get_value(self, counties: Counties) -> None: |
25 |
| - country = counties.get("England") |
| 24 | + def test_get_value(self, countries: Countries) -> None: |
| 25 | + country = countries.get("England") |
26 | 26 | if country is not None:
|
27 | 27 | assert country.code == "ENG"
|
28 | 28 | assert country.name == "England"
|
29 | 29 |
|
30 |
| - def test_get_custom_value(self, counties: Counties) -> None: |
31 |
| - country = counties["China PR"] |
| 30 | + def test_get_custom_value(self, countries: Countries) -> None: |
| 31 | + country = countries["China PR"] |
32 | 32 | assert country.code == "CHN"
|
33 | 33 | assert country.name == "China"
|
34 | 34 |
|
35 |
| - def test_get_none(self, counties: Counties) -> None: |
36 |
| - country = counties.get("foo") |
| 35 | + def test_get_none(self, countries: Countries) -> None: |
| 36 | + country = countries.get("foo") |
37 | 37 | assert country is None
|
38 | 38 |
|
39 |
| - def test_raise_keyerror(self, counties: Counties) -> None: |
| 39 | + def test_raise_keyerror(self, countries: Countries) -> None: |
40 | 40 | with pytest.raises(KeyError):
|
41 |
| - counties["foo"] |
| 41 | + countries["foo"] |
42 | 42 |
|
43 |
| - def test_search(self, counties: Counties) -> None: |
44 |
| - results = counties.search("ENG") |
| 43 | + def test_search(self, countries: Countries) -> None: |
| 44 | + results = countries.search("ENG") |
45 | 45 | assert len(results) == 3
|
46 | 46 | country = results[0]
|
47 | 47 | assert country.name == "England"
|
48 | 48 |
|
49 |
| - def test_search_limit_one(self, counties: Counties) -> None: |
50 |
| - results = counties.search("ENG", limit=1) |
| 49 | + def test_search_limit_one(self, countries: Countries) -> None: |
| 50 | + results = countries.search("ENG", limit=1) |
51 | 51 | assert len(results) == 1
|
52 | 52 |
|
53 |
| - def test_search_score_cutoff(self, counties: Counties) -> None: |
54 |
| - results = counties.search("ENG", score_cutoff=90.0) |
| 53 | + def test_search_score_cutoff(self, countries: Countries) -> None: |
| 54 | + results = countries.search("ENG", score_cutoff=90.0) |
55 | 55 | assert len(results) == 1
|
56 | 56 |
|
57 |
| - def test_search_none(self, counties: Counties) -> None: |
58 |
| - results = counties.search("foobar") |
| 57 | + def test_search_none(self, countries: Countries) -> None: |
| 58 | + results = countries.search("foobar") |
59 | 59 | assert len(results) == 0
|
60 | 60 |
|
61 |
| - def test_search_one(self, counties: Counties) -> None: |
62 |
| - country = counties.search_one("ENG") |
| 61 | + def test_search_one(self, countries: Countries) -> None: |
| 62 | + country = countries.search_one("ENG") |
63 | 63 | assert country is not None
|
64 | 64 | assert country.code == "ENG"
|
65 | 65 | assert country.name == "England"
|
66 | 66 |
|
67 |
| - def test_search_one_none(self, counties: Counties) -> None: |
68 |
| - country = counties.search_one("foobar") |
| 67 | + def test_search_one_none(self, countries: Countries) -> None: |
| 68 | + country = countries.search_one("foobar") |
69 | 69 | assert country is None
|
0 commit comments