def test_save_stations_from_page(self): path = tempfile.mkdtemp() try: count = milk.save_station_from_page(path, 1, self.CACHE_DIR) files = glob.glob(os.path.join(path, "station_*.json")) self.assertEquals(15, count) for filename in files: with open(filename) as f: d = json.load(f) self.assertEquals(os.path.join(path, "station_%d.json" % d['id']), filename) finally: shutil.rmtree(path)
def test_save_stations_from_page(self): path = tempfile.mkdtemp() try: count = milk.save_station_from_page(path, 1, self.CACHE_DIR) files = glob.glob(os.path.join(path, "station_*.json")) self.assertEquals(15, count) for filename in files: with open(filename) as f: d = json.load(f) self.assertEquals( os.path.join(path, "station_%d.json" % d['id']), filename) finally: shutil.rmtree(path)
def test_geocode_all_stations(self): path = tempfile.mkdtemp() try: station_count = milk.save_station_from_page(path, 1, self.CACHE_DIR) count = milk.geocode_station_files(self.CACHE_DIR, path) print station_count print count self.assertEquals(station_count, count) files = glob.glob(os.path.join(path, "geodata_*.json")) self.assertEquals(station_count, len(files)) for filename in files: with open(filename) as f: d = json.load(f) self.assertIsInstance(d, dict) #self.assertIn(d, 'status') finally: shutil.rmtree(path)
def test_geocode_all_stations(self): path = tempfile.mkdtemp() try: station_count = milk.save_station_from_page( path, 1, self.CACHE_DIR) count = milk.geocode_station_files(self.CACHE_DIR, path) print station_count print count self.assertEquals(station_count, count) files = glob.glob(os.path.join(path, "geodata_*.json")) self.assertEquals(station_count, len(files)) for filename in files: with open(filename) as f: d = json.load(f) self.assertIsInstance(d, dict) #self.assertIn(d, 'status') finally: shutil.rmtree(path)