github/gallery-dl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[tests] ignore 'ytdl' categories when import fails (#5095)

Browse Source
This commit is contained in:
Mike Fährmann
2024-01-21 15:16:25 +01:00
parent f3ad91b44f
commit 0d3af0d35b
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/test_extractor.py
View File

@@ -101,8 +101,14 @@ class TestExtractorModule(unittest.TestCase):
def test_categories(self): def test_categories(self):
for result in results.all(): for result in results.all():
url = result["#url"] url = result["#url"]
extr = result["#class"].from_url(url)
base, cat, sub = result["#category"] base, cat, sub = result["#category"]
try:
extr = result["#class"].from_url(url)
except ImportError as exc:
if exc.name in ("youtube_dl", "yt_dlp"):
print("Skipping '{}' category checks".format(cat))
continue
raise
self.assertEqual(extr.category, cat, url) self.assertEqual(extr.category, cat, url)
self.assertEqual(extr.subcategory, sub, url) self.assertEqual(extr.subcategory, sub, url)
self.assertEqual(extr.basecategory, base, url) self.assertEqual(extr.basecategory, base, url)