Skip to content

Commit b5d0161

Browse files
oschwaldautarch
authored andcommitted
Started updating model tests to new document format. Did iso_code and started subdivisions.
1 parent 365f933 commit b5d0161

File tree

1 file changed

+16
-23
lines changed

1 file changed

+16
-23
lines changed

tests/models_test.py

Lines changed: 16 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -27,29 +27,29 @@ def test_omni_full(self):
2727
'country': {
2828
'confidence': 99,
2929
'geoname_id': 1,
30-
'iso_3166_1_alpha_2': 'US',
31-
'iso_3166_1_alpha_3': 'USA',
30+
'iso_code': 'US',
3231
'names': {'en': 'United States of America'},
3332
},
3433
'location': {
3534
'accuracy_radius': 1500,
3635
'latitude': 44.98,
3736
'longitude': 93.2636,
3837
'metro_code': 765,
39-
'postal_code': '55401',
40-
'postal_confidence': 33,
4138
'time_zone': 'America/Chicago',
4239
},
43-
'region': {
40+
'postal': {
41+
code: '55401',
42+
confidence: 33,
43+
},
44+
'subdivisions': [{
4445
'confidence': 88,
4546
'geoname_id': 574635,
46-
'iso_3166_2': 'MN',
47+
'iso_code': 'MN',
4748
'names': {'en': 'Minnesota'},
48-
},
49+
}],
4950
'registered_country': {
5051
'geoname_id': 2,
51-
'iso_3166_1_alpha_2': 'CA',
52-
'iso_3166_1_alpha_3': 'CAN',
52+
'iso_code': 'CA',
5353
'names': {'en': 'Canada'},
5454
},
5555
'traits': {
@@ -116,14 +116,12 @@ def test_city_full(self):
116116
},
117117
'country': {
118118
'geoname_id': 1,
119-
'iso_3166_1_alpha_2': 'US',
120-
'iso_3166_1_alpha_3': 'USA',
119+
'iso_code': 'US',
121120
'names': {'en': 'United States of America'},
122121
},
123122
'registered_country': {
124123
'geoname_id': 2,
125-
'iso_3166_1_alpha_2': 'CA',
126-
'iso_3166_1_alpha_3': 'CAN',
124+
'iso_code': 'CA',
127125
'names': {'en': 'Canada'},
128126
},
129127
'traits': {
@@ -160,21 +158,17 @@ def test_city_full(self):
160158
'continent name is correct')
161159
self.assertEqual(model.country.geoname_id, 1,
162160
'country geoname_id is 1')
163-
self.assertEqual(model.country.iso_3166_1_alpha_2, 'US',
164-
'country iso_3166_1_alpha_2 is US')
165-
self.assertEqual(model.country.iso_3166_1_alpha_3, 'USA',
166-
'country iso_3166_1_alpha_3 is USA')
161+
self.assertEqual(model.country.iso_code, 'US',
162+
'country iso_code is US')
167163
self.assertEqual(model.country.names,
168164
{'en': 'United States of America'},
169165
'country names is correct')
170166
self.assertEqual(model.country.name, 'United States of America',
171167
'country name is correct')
172168
self.assertEqual(model.country.confidence, None,
173169
'country confidence is None')
174-
self.assertEqual(model.registered_country.iso_3166_1_alpha_2, 'CA',
175-
'registered_country iso_3166_1_alpha_2 is CA')
176-
self.assertEqual(model.registered_country.iso_3166_1_alpha_3, 'CAN',
177-
'registered_country iso_3166_1_alpha_3 is CAN')
170+
self.assertEqual(model.registered_country.iso_code, 'CA',
171+
'registered_country iso_code is CA')
178172
self.assertEqual(model.registered_country.names,
179173
{'en': 'Canada'},
180174
'registered_country names is correct')
@@ -201,8 +195,7 @@ def test_names(self):
201195
},
202196
'country': {
203197
'geoname_id': 1,
204-
'iso_3166_1_alpha_2': 'US',
205-
'iso_3166_1_alpha_3': 'USA',
198+
'iso_code': 'US',
206199
'names': {
207200
'en': 'United States of America',
208201
'fr': 'États-Unis',

0 commit comments

Comments
 (0)