diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_youtube_signature.py | 5 | 
1 files changed, 3 insertions, 2 deletions
| diff --git a/test/test_youtube_signature.py b/test/test_youtube_signature.py index 13d228cd8..09696e19a 100644 --- a/test/test_youtube_signature.py +++ b/test/test_youtube_signature.py @@ -8,11 +8,11 @@ import sys  import unittest  sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) -  import io  import re  import string +from test.helper import FakeYDL  from youtube_dl.extractor import YoutubeIE  from youtube_dl.compat import compat_str, compat_urlretrieve @@ -88,7 +88,8 @@ def make_tfunc(url, stype, sig_input, expected_sig):          if not os.path.exists(fn):              compat_urlretrieve(url, fn) -        ie = YoutubeIE() +        ydl = FakeYDL() +        ie = YoutubeIE(ydl)          if stype == 'js':              with io.open(fn, encoding='utf-8') as testf:                  jscode = testf.read() | 
