def test_download(self):
for city in self.test_cities:
city_data = systems.get_city_by_name(city[0], city[1])
t, failures = download.save(city[0], city[1], should_archive=True)
self.assertEqual(len(failures), 0)
file_absolute = files.get_file_path(city_data, t)
file_current = files.get_current_file_path(city_data)
self.assertTrue(os.path.exists(file_absolute))
self.assertTrue(os.path.exists(file_current))
def test_download(self):
"""
Test that downloading the data results in physical file being created
for all systems specified.
"""
for city in self.test_cities:
city_data = systems.get_city_by_name(city[0], city[1])
t, failures = download.save(city[0], city[1], should_archive=True)
self.assertEqual(len(failures), 0)
file_absolute = files.get_file_path(city_data, t)
file_current = files.get_current_file_path(city_data)
self.assertTrue(os.path.exists(file_absolute))
self.assertTrue(os.path.exists(file_current))
def test_download_create_dir(self):
# tests that script will attempt to create data directories
# if they don't exist
import shutil
city_data = {'system': 'sharengo', 'name': 'milano'}
data_dir = files.get_data_dir(city_data)
# delete if already exists
if os.path.exists(data_dir):
shutil.rmtree(data_dir)
# download
t, failures = download.save(city_data['system'], city_data['name'], False)
file_current = files.get_current_file_path(city_data)
# test it was downloaded
self.assertEqual(len(failures), 0)
self.assertTrue(os.path.exists(file_current))
def test_download_create_dir(self):
"""
Tests that downloader will attempt to create data directories
if they don't exist.
"""
import shutil
city_data = {'system': 'sharengo', 'name': 'milano'}
data_dir = files.get_data_dir(city_data)
# delete if already exists
if os.path.exists(data_dir):
shutil.rmtree(data_dir)
# download
t, failures = download.save(city_data['system'], city_data['name'], False)
file_current = files.get_current_file_path(city_data)
# test it was downloaded
self.assertEqual(len(failures), 0)
self.assertTrue(os.path.exists(file_current))
