mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-11-27 15:16:52 +01:00
parent
38a40c9e16
commit
b5ac45b197
@ -35,6 +35,9 @@ def process_info(self, info_dict):
|
|||||||
def to_screen(self, msg):
|
def to_screen(self, msg):
|
||||||
self.msgs.append(msg)
|
self.msgs.append(msg)
|
||||||
|
|
||||||
|
def dl(self, *args, **kwargs):
|
||||||
|
assert False, 'Downloader must not be invoked for test_YoutubeDL'
|
||||||
|
|
||||||
|
|
||||||
def _make_result(formats, **kwargs):
|
def _make_result(formats, **kwargs):
|
||||||
res = {
|
res = {
|
||||||
@ -117,35 +120,24 @@ def test_format_selection(self):
|
|||||||
]
|
]
|
||||||
info_dict = _make_result(formats)
|
info_dict = _make_result(formats)
|
||||||
|
|
||||||
ydl = YDL({'format': '20/47'})
|
def test(inp, *expected, multi=False):
|
||||||
|
ydl = YDL({
|
||||||
|
'format': inp,
|
||||||
|
'allow_multiple_video_streams': multi,
|
||||||
|
'allow_multiple_audio_streams': multi,
|
||||||
|
})
|
||||||
ydl.process_ie_result(info_dict.copy())
|
ydl.process_ie_result(info_dict.copy())
|
||||||
downloaded = ydl.downloaded_info_dicts[0]
|
downloaded = map(lambda x: x['format_id'], ydl.downloaded_info_dicts)
|
||||||
self.assertEqual(downloaded['format_id'], '47')
|
self.assertEqual(list(downloaded), list(expected))
|
||||||
|
|
||||||
ydl = YDL({'format': '20/71/worst'})
|
test('20/47', '47')
|
||||||
ydl.process_ie_result(info_dict.copy())
|
test('20/71/worst', '35')
|
||||||
downloaded = ydl.downloaded_info_dicts[0]
|
test(None, '2')
|
||||||
self.assertEqual(downloaded['format_id'], '35')
|
test('webm/mp4', '47')
|
||||||
|
test('3gp/40/mp4', '35')
|
||||||
ydl = YDL()
|
test('example-with-dashes', 'example-with-dashes')
|
||||||
ydl.process_ie_result(info_dict.copy())
|
test('all', '35', 'example-with-dashes', '45', '47', '2') # Order doesn't actually matter for this
|
||||||
downloaded = ydl.downloaded_info_dicts[0]
|
test('mergeall', '2+47+45+example-with-dashes+35', multi=True)
|
||||||
self.assertEqual(downloaded['format_id'], '2')
|
|
||||||
|
|
||||||
ydl = YDL({'format': 'webm/mp4'})
|
|
||||||
ydl.process_ie_result(info_dict.copy())
|
|
||||||
downloaded = ydl.downloaded_info_dicts[0]
|
|
||||||
self.assertEqual(downloaded['format_id'], '47')
|
|
||||||
|
|
||||||
ydl = YDL({'format': '3gp/40/mp4'})
|
|
||||||
ydl.process_ie_result(info_dict.copy())
|
|
||||||
downloaded = ydl.downloaded_info_dicts[0]
|
|
||||||
self.assertEqual(downloaded['format_id'], '35')
|
|
||||||
|
|
||||||
ydl = YDL({'format': 'example-with-dashes'})
|
|
||||||
ydl.process_ie_result(info_dict.copy())
|
|
||||||
downloaded = ydl.downloaded_info_dicts[0]
|
|
||||||
self.assertEqual(downloaded['format_id'], 'example-with-dashes')
|
|
||||||
|
|
||||||
def test_format_selection_audio(self):
|
def test_format_selection_audio(self):
|
||||||
formats = [
|
formats = [
|
||||||
|
@ -1758,6 +1758,7 @@ def _merge(formats_pair):
|
|||||||
def _check_formats(formats):
|
def _check_formats(formats):
|
||||||
if not check_formats:
|
if not check_formats:
|
||||||
yield from formats
|
yield from formats
|
||||||
|
return
|
||||||
for f in formats:
|
for f in formats:
|
||||||
self.to_screen('[info] Testing format %s' % f['format_id'])
|
self.to_screen('[info] Testing format %s' % f['format_id'])
|
||||||
temp_file = tempfile.NamedTemporaryFile(
|
temp_file = tempfile.NamedTemporaryFile(
|
||||||
|
Loading…
Reference in New Issue
Block a user