def test_download_file_gdrive():
files = datautils.download_file(name='requirements.txt',
url='1_K6gqtZrvGWsINfkY0J6Evbd3bbWLg_0',
cache_dir=BASE_PATH,
extract=False)
print(files)
assert len(files) > 0
for f in files:
os.remove(f)
def test_download_archive_gdrive():
files = datautils.download_file(name='test.tar.gz',
url='1Sr1fm8PaaKQuvVpl34t6xyWty61o2psH',
cache_dir=BASE_PATH)
print(files)
assert len(files) > 0
for f in files:
if f.parent.exists():
shutil.rmtree(f.parent)
def test_download_file_github():
files = datautils.download_file(
name='densenet121_weights_tf_dim_ordering_tf_kernels.h5',
url=
'https://github.com/keras-team/keras-applications/releases/download/densenet/densenet121_weights_tf_dim_ordering_tf_kernels_notop.h5',
cache_dir=BASE_PATH,
extract=False)
print(files)
assert len(files) > 0
shutil.rmtree(files[0].parent)
def test_download_archive():
files = datautils.download_file(
name='mnist.zip',
url='https://github.com/mlampros/DataSets/raw/master/mnist.zip',
cache_dir=BASE_PATH,
extract=True)
print(files)
assert len(files) > 0
shutil.rmtree(files[0].parent)
shutil.rmtree(files[0].parent.parent)
def test_download_file_extract():
with pytest.raises(AssertionError):
datautils.download_file(name='requirements.txt',
url='1_K6gqtZrvGWsINfkY0J6Evbd3bbWLg_0',
cache_dir=BASE_PATH)
