commit
1c1218fefc
3 changed files with 148 additions and 8 deletions
@ -0,0 +1,49 @@ |
|||||||
|
#!/usr/bin/env python |
||||||
|
|
||||||
|
import sys |
||||||
|
import unittest |
||||||
|
|
||||||
|
# Allow direct execution |
||||||
|
import os |
||||||
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) |
||||||
|
|
||||||
|
from helper import FakeYDL, parameters |
||||||
|
|
||||||
|
class YDL(FakeYDL): |
||||||
|
def __init__(self): |
||||||
|
super(YDL, self).__init__() |
||||||
|
self.downloaded_info_dicts = [] |
||||||
|
def process_info(self, info_dict): |
||||||
|
self.downloaded_info_dicts.append(info_dict) |
||||||
|
|
||||||
|
class TestFormatSelection(unittest.TestCase): |
||||||
|
def test_prefer_free_formats(self): |
||||||
|
# Same resolution => download webm |
||||||
|
ydl = YDL() |
||||||
|
ydl.params['prefer_free_formats'] = True |
||||||
|
formats = [{u'ext': u'webm', u'height': 460},{u'ext': u'mp4', u'height': 460}] |
||||||
|
info_dict = {u'formats': formats, u'extractor': u'test'} |
||||||
|
ydl.process_ie_result(info_dict) |
||||||
|
downloaded = ydl.downloaded_info_dicts[0] |
||||||
|
self.assertEqual(downloaded[u'ext'], u'webm') |
||||||
|
|
||||||
|
# Different resolution => download best quality (mp4) |
||||||
|
ydl = YDL() |
||||||
|
ydl.params['prefer_free_formats'] = True |
||||||
|
formats = [{u'ext': u'webm', u'height': 720},{u'ext': u'mp4',u'height': 1080}] |
||||||
|
info_dict[u'formats'] = formats |
||||||
|
ydl.process_ie_result(info_dict) |
||||||
|
downloaded = ydl.downloaded_info_dicts[0] |
||||||
|
self.assertEqual(downloaded[u'ext'], u'mp4') |
||||||
|
|
||||||
|
# No prefer_free_formats => keep original formats order |
||||||
|
ydl = YDL() |
||||||
|
ydl.params['prefer_free_formats'] = False |
||||||
|
formats = [{u'ext': u'webm', u'height': 720},{u'ext': u'flv',u'height': 720}] |
||||||
|
info_dict[u'formats'] = formats |
||||||
|
ydl.process_ie_result(info_dict) |
||||||
|
downloaded = ydl.downloaded_info_dicts[0] |
||||||
|
self.assertEqual(downloaded[u'ext'], u'flv') |
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
unittest.main() |
Loading…
Reference in new issue