diff --git a/test/test_HaruhiDL.py b/test/test_HaruhiDL.py index b35b16b28..c6346118a 100644 --- a/test/test_HaruhiDL.py +++ b/test/test_HaruhiDL.py @@ -14,7 +14,10 @@ import copy from test.helper import FakeHDL, assertRegexpMatches from haruhi_dl import HaruhiDL from haruhi_dl.compat import compat_str, compat_urllib_error -from haruhi_dl.extractor import YoutubeIE + +# extractors must be imported directly from their files, +# __init__.py points to lazy_extractors, which have no additional attributes/methods +from haruhi_dl.extractor.youtube import YoutubeIE from haruhi_dl.extractor.common import InfoExtractor from haruhi_dl.postprocessor.common import PostProcessor from haruhi_dl.utils import ExtractorError, match_filter_func diff --git a/test/test_all_urls.py b/test/test_all_urls.py index 739bd3345..0913df476 100644 --- a/test/test_all_urls.py +++ b/test/test_all_urls.py @@ -12,11 +12,12 @@ sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) from test.helper import gettestcases -from haruhi_dl.extractor import ( - FacebookIE, - gen_extractors, - YoutubeIE, -) +from haruhi_dl.extractor import gen_extractors + +# extractors must be imported directly from their files, +# __init__.py points to lazy_extractors, which have no additional attributes/methods +from haruhi_dl.extractor.youtube import YoutubeIE +from haruhi_dl.extractor.facebook import FacebookIE class TestAllURLsMatching(unittest.TestCase):