Allow moving tests into IE files

[youtube-dl] / test / test_download.py

diff --git a/test/test_download.py b/test/test_download.py
index 067bde4bb9159f4d56ae34bbc4690a8425a11550..0428a79fe7450e0fdbd97cc698c243fa76d9ea6e 100644 (file)
--- a/test/test_download.py
+++ b/test/test_download.py
@@ -58,6 +58,13 @@ def _file_md5(fn):
 
 with io.open(DEF_FILE, encoding='utf-8') as deff:
     defs = json.load(deff)
+for ie in youtube_dl.extractor.gen_extractors():
+    t = getattr(ie, '_TEST', None)
+    if t:
+        t['name'] = type(ie).__name__[:-len('IE')]
+        defs.append(t)
+
+
 with io.open(PARAMETERS_FILE, encoding='utf-8') as pf:
     parameters = json.load(pf)