]>
jfr.im git - yt-dlp.git/blob - test/test_youtube_lists.py
2 from __future__
import unicode_literals
4 # Allow direct execution
8 sys
.path
.insert(0, os
.path
.dirname(os
.path
.dirname(os
.path
.abspath(__file__
))))
10 from test
.helper
import FakeYDL
, is_download_test
13 from yt_dlp
.extractor
import (
21 class TestYoutubeLists(unittest
.TestCase
):
22 def assertIsPlaylist(self
, info
):
23 """Make sure the info has '_type' set to 'playlist'"""
24 self
.assertEqual(info
['_type'], 'playlist')
26 def test_youtube_playlist_noplaylist(self
):
28 dl
.params
['noplaylist'] = True
29 ie
= YoutubePlaylistIE(dl
)
30 result
= ie
.extract('https://www.youtube.com/watch?v=FXxLjLQi3Fg&list=PLwiyx1dc3P2JR9N8gQaQN_BCvlSlap7re')
31 self
.assertEqual(result
['_type'], 'url')
32 self
.assertEqual(YoutubeIE().extract_id(result
['url']), 'FXxLjLQi3Fg')
34 def test_youtube_course(self
):
36 ie
= YoutubePlaylistIE(dl
)
37 # TODO find a > 100 (paginating?) videos course
38 result
= ie
.extract('https://www.youtube.com/course?list=ECUl4u3cNGP61MdtwGTqZA0MreSaDybji8')
39 entries
= list(result
['entries'])
40 self
.assertEqual(YoutubeIE().extract_id(entries
[0]['url']), 'j9WZyLZCBzs')
41 self
.assertEqual(len(entries
), 25)
42 self
.assertEqual(YoutubeIE().extract_id(entries
[-1]['url']), 'rYefUsYuEp0')
44 def test_youtube_mix(self
):
46 ie
= YoutubePlaylistIE(dl
)
47 result
= ie
.extract('https://www.youtube.com/watch?v=W01L70IGBgE&index=2&list=RDOQpdSVF_k_w')
48 entries
= result
['entries']
49 self
.assertTrue(len(entries
) >= 50)
50 original_video
= entries
[0]
51 self
.assertEqual(original_video
['id'], 'OQpdSVF_k_w')
53 def test_youtube_toptracks(self
):
54 print('Skipping: The playlist page gives error 500')
57 ie
= YoutubePlaylistIE(dl
)
58 result
= ie
.extract('https://www.youtube.com/playlist?list=MCUS')
59 entries
= result
['entries']
60 self
.assertEqual(len(entries
), 100)
62 def test_youtube_flat_playlist_extraction(self
):
64 dl
.params
['extract_flat'] = True
66 result
= ie
.extract('https://www.youtube.com/playlist?list=PL4lCao7KL_QFVb7Iudeipvc2BCavECqzc')
67 self
.assertIsPlaylist(result
)
68 entries
= list(result
['entries'])
69 self
.assertTrue(len(entries
) == 1)
71 self
.assertEqual(video
['_type'], 'url_transparent')
72 self
.assertEqual(video
['ie_key'], 'Youtube')
73 self
.assertEqual(video
['id'], 'BaW_jenozKc')
74 self
.assertEqual(video
['url'], 'BaW_jenozKc')
75 self
.assertEqual(video
['title'], 'youtube-dl test video "\'/\\ä↭𝕐')
76 self
.assertEqual(video
['duration'], 10)
77 self
.assertEqual(video
['uploader'], 'Philipp Hagemeister')
80 if __name__
== '__main__':