class GeonamesCacheTestSuite(unittest.TestCase):
"""GeonamesCache test cases."""
def setUp(self):
self.geonamescache = GeonamesCache()
def test_continents(self):
continents = self.geonamescache.get_continents()
testdata = (('AF', 'Africa'), ('AN', 'Antarctica'), ('AS', 'Asia'),
('EU', 'Europe'), ('NA', 'North America'),
('OC', 'Oceania'), ('SA', 'South America'))
for code, name in testdata:
self.assertTrue(code in continents)
self.assertEqual(name, continents[code]['name'])
for code in ['XX', 'OO']:
self.assertTrue(code not in continents)
def test_get_countries(self):
countries = self.geonamescache.get_countries()
testdata = (('ES', 'Spain'), ('FR', 'France'), ('US', 'United States'))
for code, name in testdata:
self.assertTrue(code in countries)
self.assertEqual(name, countries[code]['name'])
for code in ['XX', 'OO']:
self.assertTrue(code not in countries)
def test_us_states(self):
us_states = self.geonamescache.get_us_states()
testdata = (('NM', 'New Mexico'), ('CA', 'California'), ('NV',
'Nevada'))
for code, name in testdata:
self.assertTrue(code in us_states)
self.assertEqual(name, us_states[code]['name'])
for code in ['XX', 'OO']:
self.assertTrue(code not in us_states)
def test_get_countries_by_names(self):
# Length of get_countries_by_names dict and get_countries dict must be
# the same, unless country names wouldn't be unique.
self.assertTrue(len(self.geonamescache.get_countries_by_names()),
len(self.geonamescache.get_countries()))
def test_get_cities_by_name(self):
cities = self.geonamescache.get_cities()
for gid, name in (('3191316', 'Samobor'), ('3107112',
'Rivas-Vaciamadrid')):
self.assertEqual(name, cities[gid]['name'])
def test_get_cities_by_name_madrid(self):
self.assertEqual(2,
len(self.geonamescache.get_cities_by_name('Madrid')))
def test_us_counties_len(self):
# Make sure there are 3235 counties, which includes Puerto Rico etc.
us_counties = self.geonamescache.get_us_counties()
self.assertEqual(3235, len(us_counties))
# This is by far the ugliest Python script I ever wrote
# withoud cities 14954 location string remain unresovled
# with unique city names 6345 (number is not up to date) location string remain unresovled
# with largest city name 4623 location string remain unresovled
# with manually resolved locations 3333 location string remain unresovled
import csv, json, re
from geonamescache import GeonamesCache
from loclists import check_unresolved
unresolved_locations = []
commits_by_countries = {}
countries_by_locstr = {}
gc = GeonamesCache()
countries = gc.get_countries()
countries_by_names = gc.get_countries_by_names()
us_states = gc.get_us_states()
us_states_by_names = gc.get_us_states_by_names()
re_ignore = re.compile(r'[\.\(\)\d-]')
re_ws = re.compile(r'\s{2,}')
def test_locs(locs):
for loc in locs:
loc = loc.strip().lower()
loctitle = loc.title()
locupper = loc.upper()
if loc in countries_by_names:
return loc
elif loctitle in countries_by_names:
class GeonamesCacheTestSuite(unittest.TestCase):
"""GeonamesCache test cases."""
def setUp(self):
self.geonamescache = GeonamesCache()
def test_continents(self):
continents = self.geonamescache.get_continents()
testdata = (
('AF', 'Africa'),
('AN', 'Antarctica'),
('AS', 'Asia'),
('EU', 'Europe'),
('NA', 'North America'),
('OC', 'Oceania'),
('SA', 'South America')
)
for code, name in testdata:
self.assertTrue(code in continents)
self.assertEqual(name, continents[code]['name'])
for code in ['XX', 'OO']:
self.assertTrue(code not in continents)
def test_get_countries(self):
countries = self.geonamescache.get_countries()
testdata = (('ES', 'Spain'), ('FR', 'France'), ('US', 'United States'))
for code, name in testdata:
self.assertTrue(code in countries)
self.assertEqual(name, countries[code]['name'])
for code in ['XX', 'OO']:
self.assertTrue(code not in countries)
def test_us_states(self):
us_states = self.geonamescache.get_us_states()
testdata = (
('NM', 'New Mexico'), ('CA', 'California'), ('NV', 'Nevada'))
for code, name in testdata:
self.assertTrue(code in us_states)
self.assertEqual(name, us_states[code]['name'])
for code in ['XX', 'OO']:
self.assertTrue(code not in us_states)
def test_get_countries_by_names(self):
# Length of get_countries_by_names dict and get_countries dict must be
# the same, unless country names wouldn't be unique.
self.assertTrue(len(self.geonamescache.get_countries_by_names()),
len(self.geonamescache.get_countries()))
def test_get_cities_by_name(self):
cities = self.geonamescache.get_cities()
for gid, name in (('3191316', 'Samobor'), ('3107112', 'Rivas-Vaciamadrid')):
self.assertEqual(name, cities[gid]['name'])
def test_get_cities_by_name_madrid(self):
self.assertEqual(
2, len(self.geonamescache.get_cities_by_name('Madrid')))
def test_us_counties_len(self):
# Make sure there are 3235 counties, which includes Puerto Rico etc.
us_counties = self.geonamescache.get_us_counties()
self.assertEqual(3234, len(us_counties))
# This is by far the ugliest Python script I ever wrote
# withoud cities 14954 location string remain unresovled
# with unique city names 6345 (number is not up to date) location string remain unresovled
# with largest city name 4623 location string remain unresovled
# with manually resolved locations 3333 location string remain unresovled
import csv, json, re
from geonamescache import GeonamesCache
from loclists import check_unresolved
unresolved_locations = []
commits_by_countries = {}
countries_by_locstr = {}
gc = GeonamesCache()
countries = gc.get_countries()
countries_by_names = gc.get_countries_by_names()
us_states = gc.get_us_states()
us_states_by_names = gc.get_us_states_by_names()
re_ignore = re.compile(r'[\.\(\)\d-]')
re_ws = re.compile(r'\s{2,}')
def test_locs(locs):
for loc in locs:
loc = loc.strip().lower()
loctitle = loc.title()
locupper = loc.upper()
if loc in countries_by_names:
return loc
elif loctitle in countries_by_names:
