Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 6 additions & 6 deletions tests/test_google_images_download.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,35 +15,35 @@ def silent_remove_of_file(file):

def test_download_images_to_default_location():
start_time = time.time()
argumnets = {
arguments = {
"keywords": "Polar bears",
"limit": 5,
"print_urls": False
}
try:
temp = argumnets['output_folder']
temp = arguments['output_folder']
except KeyError:
pass
else:
assert False, "This test checks download to default location yet an output folder was provided"

output_folder_path = os.path.join(os.path.realpath('.'), 'downloads', '{}'.format(argumnets['keywords']))
output_folder_path = os.path.join(os.path.realpath('.'), 'downloads', '{}'.format(arguments['keywords']))
if os.path.exists(output_folder_path):
start_amount_of_files_in_output_folder = len([name for name in os.listdir(output_folder_path) if os.path.isfile(os.path.join(output_folder_path, name)) and os.path.getctime(os.path.join(output_folder_path, name)) < start_time])
else:
start_amount_of_files_in_output_folder = 0

response = google_images_download.googleimagesdownload()
response.download(argumnets)
response.download(arguments)
files_modified_after_test_started = [name for name in os.listdir(output_folder_path) if os.path.isfile(os.path.join(output_folder_path, name)) and os.path.getmtime(os.path.join(output_folder_path, name)) > start_time]
end_amount_of_files_in_output_folder = len(files_modified_after_test_started)
print(f"Files downloaded by test {__name__}:")
for file in files_modified_after_test_started:
print(os.path.join(output_folder_path, file))


# assert end_amount_of_files_in_output_folder - start_amount_of_files_in_output_folder == argumnets['limit']
assert end_amount_of_files_in_output_folder == argumnets['limit']
# assert end_amount_of_files_in_output_folder - start_amount_of_files_in_output_folder == arguments['limit']
assert end_amount_of_files_in_output_folder == arguments['limit']

print(f"Cleaning up all files downloaded by test {__name__}...")
for file in files_modified_after_test_started:
Expand Down