fix tests

This commit is contained in:
msramalho
2022-06-21 01:21:53 +02:00
parent e1e3648852
commit c25880ee6d

View File

@@ -135,4 +135,8 @@ def test_scrape_video_only2():
with tempfile.TemporaryDirectory(dir="./") as tempdir:
vks.download_media(res, tempdir)
found_files = set(os.listdir(tempdir))
assert "video-17546758_456239898_0.webm" in found_files
# different systems might attribute different extension
assert (
"video-17546758_456239898_0.webm" in found_files
or "video-17546758_456239898_0.mp4" in found_files
)