General: Fix test.

This commit is contained in:
Dylan Araps 2017-06-27 21:28:12 +10:00
parent b5c27de8b5
commit d82de89d36

View File

@ -25,7 +25,7 @@ def random_img(img_dir):
images = [img for img in os.scandir(img_dir)
if img.name.endswith(file_types) and img.name != current_wall]
return pathlib.Path(img_dir / random.choice(images))
return pathlib.Path(img_dir / random.choice(images).name)
def get_image(img):