diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_download.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/test/test_download.py b/test/test_download.py index a8de1d002..59a6e1498 100644 --- a/test/test_download.py +++ b/test/test_download.py @@ -81,9 +81,8 @@ def generator(test_case): params.update(test_case.get('params', {})) fd = FileDownloader(params) - fd.add_info_extractor(ie()) - for ien in test_case.get('add_ie', []): - fd.add_info_extractor(getattr(youtube_dl.InfoExtractors, ien + 'IE')()) + for ie in youtube_dl.InfoExtractors.gen_extractors(): + fd.add_info_extractor(ie) finished_hook_called = set() def _hook(status): if status['status'] == 'finished': @@ -103,7 +102,7 @@ def generator(test_case): if retry == RETRIES: raise # Check if the exception is not a network related one - if not err.exc_info[0] in (ZeroDivisionError, compat_urllib_error.URLError, socket.timeout): + if not err.exc_info[0] in (compat_urllib_error.URLError, socket.timeout, UnavailableVideoError): raise print('Retrying: {0} failed tries\n\n##########\n\n'.format(retry)) |