def test_download_file(self):
""""""
client = Client()
client.MAX_WAITING_PERIOD = 0.1
dataset = client.get_dataset_info(name='FAOSTAT_data_producer_prices',
status='test_dataset')
# test failure
def fail(x, y):
raise Download.Failed("on purpose")
self.assertRaises(Download.Failed,
client._download_file,
DATA_DIR,
dataset.files[0],
check=fail)
self.assertFalse(
DATA_DIR.joinpath(dataset.files[0].file_name).is_file())
# test success
download = client._download_file(DATA_DIR, dataset.files[0])
self.assertEqual(download, DATA_DIR / dataset.files[0].file_name)
self.assertTrue(download.is_file())
self.assertEqual(download.stat().st_size, dataset.files[0].file_size)
download.unlink()
self.assertFalse(download.is_file())
def test_download_dataset(self):
""""""
client = Client()
client.MAX_WAITING_PERIOD = 0.1
dataset = client.get_dataset(name='test_write_raster')
downloads = client.download_dataset(dataset, target_dir=DATA_DIR)
self.assertEqual(len(downloads), 2)
for download, dsfile in zip(downloads, dataset.files):
self.assertTrue(download.is_file())
self.assertEqual(download.stat().st_size, dsfile.file_size)
self.assertEqual(download.name, dsfile.file_name)
self.assertEqual(download.parent.name, dataset.version)
self.assertEqual(download.parent.parent.name, dataset.name)
self.assertEqual(download.parent.parent.parent.name, dataset.data_type.data_type)
download.unlink()
rm_empty_dir(download.parent.parent.parent)
