Mercurial > hg > GlobalNeighbors
diff tests/test_read.py @ 3:49aae0c0293b
improved test coverage
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Sat, 24 Jun 2017 14:48:31 -0700 |
parents | 1b94f3bf97e5 |
children |
line wrap: on
line diff
--- a/tests/test_read.py Sat Jun 24 14:04:00 2017 -0700 +++ b/tests/test_read.py Sat Jun 24 14:48:31 2017 -0700 @@ -7,7 +7,9 @@ import os import unittest from globalneighbors import schema +from globalneighbors.locations import locations from globalneighbors.read import read_tsv +from globalneighbors.read import read_cities from globalneighbors.read import read_city_list @@ -52,6 +54,19 @@ """ensure we can read the cities as unicode""" cities = read_city_list(self.full_tsv) + for city in cities: + for field in schema.unicode_fields: + assert isinstance(city[field], unicode) + + def test_iterative_locations(self): + """assert we can read into locations as a generator""" + + with open(self.test_tsv) as f: + cities = locations(read_cities(f)) + for geonameid, (lat, lon) in cities.items(): + assert -90. <= lat <= 90. + assert -180. <= lon <= 180. + assert type(geonameid) == schema.types['geonameid'] if __name__ == '__main__':