def test_download_and_extract_movielens(size, tmp):
"""Test movielens data download and extract
"""
zip_path = os.path.join(tmp, "ml.zip")
download_movielens(size, dest_path=zip_path)
assert len(os.listdir(tmp)) == 1
assert os.path.exists(zip_path)
rating_path = os.path.join(tmp, "rating.dat")
item_path = os.path.join(tmp, "item.dat")
extract_movielens(
size, rating_path=rating_path, item_path=item_path, zip_path=zip_path
)
# Test if raw-zip file, rating file, and item file are cached
assert len(os.listdir(tmp)) == 3
assert os.path.exists(rating_path)
assert os.path.exists(item_path)
def test_download_and_extract_movielens(size, tmp):
"""Test movielens data download and extract"""
zip_path = os.path.join(tmp, "ml.zip")
download_movielens(size, dest_path=zip_path)
assert len(os.listdir(tmp)) == 1
assert os.path.exists(zip_path)
rating_path = os.path.join(tmp, "rating.dat")
item_path = os.path.join(tmp, "item.dat")
extract_movielens(size,
rating_path=rating_path,
item_path=item_path,
zip_path=zip_path)
# Test if raw-zip file, rating file, and item file are cached
assert len(os.listdir(tmp)) == 3
assert os.path.exists(rating_path)
assert os.path.exists(item_path)
def test_download_and_extract_movielens(size):
"""Test movielens data download and extract
"""
with TemporaryDirectory() as tmp_dir:
zip_path = os.path.join(tmp_dir, "ml.zip")
download_movielens(size, dest_path=zip_path)
assert len(os.listdir(tmp_dir)) == 1
assert os.path.exists(zip_path)
rating_path = os.path.join(tmp_dir, "rating.dat")
item_path = os.path.join(tmp_dir, "item.dat")
extract_movielens(size,
rating_path=rating_path,
item_path=item_path,
zip_path=zip_path)
assert len(os.listdir(tmp_dir)) == 3
assert os.path.exists(rating_path)
assert os.path.exists(item_path)
