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__':