]>
jfr.im git - yt-dlp.git/blob - test/test_subtitles.py
3 # Allow direct execution
7 sys
.path
.insert(0, os
.path
.dirname(os
.path
.dirname(os
.path
.abspath(__file__
))))
9 from test
.helper
import FakeYDL
, md5
12 from youtube_dl
.extractor
import (
22 class BaseTestSubtitles(unittest
.TestCase
):
27 self
.ie
= self
.IE(self
.DL
)
29 def getInfoDict(self
):
30 info_dict
= self
.ie
.extract(self
.url
)
33 def getSubtitles(self
):
34 info_dict
= self
.getInfoDict()
35 return info_dict
['subtitles']
38 class TestYoutubeSubtitles(BaseTestSubtitles
):
42 def test_youtube_no_writesubtitles(self
):
43 self
.DL
.params
['writesubtitles'] = False
44 subtitles
= self
.getSubtitles()
45 self
.assertEqual(subtitles
, None)
47 def test_youtube_subtitles(self
):
48 self
.DL
.params
['writesubtitles'] = True
49 subtitles
= self
.getSubtitles()
50 self
.assertEqual(md5(subtitles
['en']), '4cd9278a35ba2305f47354ee13472260')
52 def test_youtube_subtitles_lang(self
):
53 self
.DL
.params
['writesubtitles'] = True
54 self
.DL
.params
['subtitleslangs'] = ['it']
55 subtitles
= self
.getSubtitles()
56 self
.assertEqual(md5(subtitles
['it']), '164a51f16f260476a05b50fe4c2f161d')
58 def test_youtube_allsubtitles(self
):
59 self
.DL
.params
['writesubtitles'] = True
60 self
.DL
.params
['allsubtitles'] = True
61 subtitles
= self
.getSubtitles()
62 self
.assertEqual(len(subtitles
.keys()), 13)
64 def test_youtube_subtitles_sbv_format(self
):
65 self
.DL
.params
['writesubtitles'] = True
66 self
.DL
.params
['subtitlesformat'] = 'sbv'
67 subtitles
= self
.getSubtitles()
68 self
.assertEqual(md5(subtitles
['en']), '13aeaa0c245a8bed9a451cb643e3ad8b')
70 def test_youtube_subtitles_vtt_format(self
):
71 self
.DL
.params
['writesubtitles'] = True
72 self
.DL
.params
['subtitlesformat'] = 'vtt'
73 subtitles
= self
.getSubtitles()
74 self
.assertEqual(md5(subtitles
['en']), '3cb210999d3e021bd6c7f0ea751eab06')
76 def test_youtube_list_subtitles(self
):
77 self
.DL
.expect_warning(u
'Video doesn\'t have automatic captions')
78 self
.DL
.params
['listsubtitles'] = True
79 info_dict
= self
.getInfoDict()
80 self
.assertEqual(info_dict
, None)
82 def test_youtube_automatic_captions(self
):
83 self
.url
= '8YoUxe5ncPo'
84 self
.DL
.params
['writeautomaticsub'] = True
85 self
.DL
.params
['subtitleslangs'] = ['it']
86 subtitles
= self
.getSubtitles()
87 self
.assertTrue(subtitles
['it'] is not None)
89 def test_youtube_nosubtitles(self
):
90 self
.DL
.expect_warning(u
'video doesn\'t have subtitles')
91 self
.url
= 'n5BB19UTcdA'
92 self
.DL
.params
['writesubtitles'] = True
93 self
.DL
.params
['allsubtitles'] = True
94 subtitles
= self
.getSubtitles()
95 self
.assertEqual(len(subtitles
), 0)
97 def test_youtube_multiple_langs(self
):
98 self
.url
= 'QRS8MkLhQmM'
99 self
.DL
.params
['writesubtitles'] = True
100 langs
= ['it', 'fr', 'de']
101 self
.DL
.params
['subtitleslangs'] = langs
102 subtitles
= self
.getSubtitles()
104 self
.assertTrue(subtitles
.get(lang
) is not None, u
'Subtitles for \'%s\' not extracted' % lang
)
107 class TestDailymotionSubtitles(BaseTestSubtitles
):
108 url
= 'http://www.dailymotion.com/video/xczg00'
111 def test_no_writesubtitles(self
):
112 subtitles
= self
.getSubtitles()
113 self
.assertEqual(subtitles
, None)
115 def test_subtitles(self
):
116 self
.DL
.params
['writesubtitles'] = True
117 subtitles
= self
.getSubtitles()
118 self
.assertEqual(md5(subtitles
['en']), '976553874490cba125086bbfea3ff76f')
120 def test_subtitles_lang(self
):
121 self
.DL
.params
['writesubtitles'] = True
122 self
.DL
.params
['subtitleslangs'] = ['fr']
123 subtitles
= self
.getSubtitles()
124 self
.assertEqual(md5(subtitles
['fr']), '594564ec7d588942e384e920e5341792')
126 def test_allsubtitles(self
):
127 self
.DL
.params
['writesubtitles'] = True
128 self
.DL
.params
['allsubtitles'] = True
129 subtitles
= self
.getSubtitles()
130 self
.assertEqual(len(subtitles
.keys()), 5)
132 def test_list_subtitles(self
):
133 self
.DL
.expect_warning(u
'Automatic Captions not supported by this server')
134 self
.DL
.params
['listsubtitles'] = True
135 info_dict
= self
.getInfoDict()
136 self
.assertEqual(info_dict
, None)
138 def test_automatic_captions(self
):
139 self
.DL
.expect_warning(u
'Automatic Captions not supported by this server')
140 self
.DL
.params
['writeautomaticsub'] = True
141 self
.DL
.params
['subtitleslang'] = ['en']
142 subtitles
= self
.getSubtitles()
143 self
.assertTrue(len(subtitles
.keys()) == 0)
145 def test_nosubtitles(self
):
146 self
.DL
.expect_warning(u
'video doesn\'t have subtitles')
147 self
.url
= 'http://www.dailymotion.com/video/x12u166_le-zapping-tele-star-du-08-aout-2013_tv'
148 self
.DL
.params
['writesubtitles'] = True
149 self
.DL
.params
['allsubtitles'] = True
150 subtitles
= self
.getSubtitles()
151 self
.assertEqual(len(subtitles
), 0)
153 def test_multiple_langs(self
):
154 self
.DL
.params
['writesubtitles'] = True
155 langs
= ['es', 'fr', 'de']
156 self
.DL
.params
['subtitleslangs'] = langs
157 subtitles
= self
.getSubtitles()
159 self
.assertTrue(subtitles
.get(lang
) is not None, u
'Subtitles for \'%s\' not extracted' % lang
)
162 class TestTedSubtitles(BaseTestSubtitles
):
163 url
= 'http://www.ted.com/talks/dan_dennett_on_our_consciousness.html'
166 def test_no_writesubtitles(self
):
167 subtitles
= self
.getSubtitles()
168 self
.assertEqual(subtitles
, None)
170 def test_subtitles(self
):
171 self
.DL
.params
['writesubtitles'] = True
172 subtitles
= self
.getSubtitles()
173 self
.assertEqual(md5(subtitles
['en']), '4262c1665ff928a2dada178f62cb8d14')
175 def test_subtitles_lang(self
):
176 self
.DL
.params
['writesubtitles'] = True
177 self
.DL
.params
['subtitleslangs'] = ['fr']
178 subtitles
= self
.getSubtitles()
179 self
.assertEqual(md5(subtitles
['fr']), '66a63f7f42c97a50f8c0e90bc7797bb5')
181 def test_allsubtitles(self
):
182 self
.DL
.params
['writesubtitles'] = True
183 self
.DL
.params
['allsubtitles'] = True
184 subtitles
= self
.getSubtitles()
185 self
.assertTrue(len(subtitles
.keys()) >= 28)
187 def test_list_subtitles(self
):
188 self
.DL
.expect_warning(u
'Automatic Captions not supported by this server')
189 self
.DL
.params
['listsubtitles'] = True
190 info_dict
= self
.getInfoDict()
191 self
.assertEqual(info_dict
, None)
193 def test_automatic_captions(self
):
194 self
.DL
.expect_warning(u
'Automatic Captions not supported by this server')
195 self
.DL
.params
['writeautomaticsub'] = True
196 self
.DL
.params
['subtitleslang'] = ['en']
197 subtitles
= self
.getSubtitles()
198 self
.assertTrue(len(subtitles
.keys()) == 0)
200 def test_multiple_langs(self
):
201 self
.DL
.params
['writesubtitles'] = True
202 langs
= ['es', 'fr', 'de']
203 self
.DL
.params
['subtitleslangs'] = langs
204 subtitles
= self
.getSubtitles()
206 self
.assertTrue(subtitles
.get(lang
) is not None, u
'Subtitles for \'%s\' not extracted' % lang
)
209 class TestBlipTVSubtitles(BaseTestSubtitles
):
210 url
= 'http://blip.tv/a/a-6603250'
213 def test_list_subtitles(self
):
214 self
.DL
.expect_warning(u
'Automatic Captions not supported by this server')
215 self
.DL
.params
['listsubtitles'] = True
216 info_dict
= self
.getInfoDict()
217 self
.assertEqual(info_dict
, None)
219 def test_allsubtitles(self
):
220 self
.DL
.expect_warning(u
'Automatic Captions not supported by this server')
221 self
.DL
.params
['writesubtitles'] = True
222 self
.DL
.params
['allsubtitles'] = True
223 subtitles
= self
.getSubtitles()
224 self
.assertEqual(set(subtitles
.keys()), set(['en']))
225 self
.assertEqual(md5(subtitles
['en']), '5b75c300af65fe4476dff79478bb93e4')
228 class TestVimeoSubtitles(BaseTestSubtitles
):
229 url
= 'http://vimeo.com/76979871'
232 def test_no_writesubtitles(self
):
233 subtitles
= self
.getSubtitles()
234 self
.assertEqual(subtitles
, None)
236 def test_subtitles(self
):
237 self
.DL
.params
['writesubtitles'] = True
238 subtitles
= self
.getSubtitles()
239 self
.assertEqual(md5(subtitles
['en']), '8062383cf4dec168fc40a088aa6d5888')
241 def test_subtitles_lang(self
):
242 self
.DL
.params
['writesubtitles'] = True
243 self
.DL
.params
['subtitleslangs'] = ['fr']
244 subtitles
= self
.getSubtitles()
245 self
.assertEqual(md5(subtitles
['fr']), 'b6191146a6c5d3a452244d853fde6dc8')
247 def test_allsubtitles(self
):
248 self
.DL
.params
['writesubtitles'] = True
249 self
.DL
.params
['allsubtitles'] = True
250 subtitles
= self
.getSubtitles()
251 self
.assertEqual(set(subtitles
.keys()), set(['de', 'en', 'es', 'fr']))
253 def test_list_subtitles(self
):
254 self
.DL
.expect_warning(u
'Automatic Captions not supported by this server')
255 self
.DL
.params
['listsubtitles'] = True
256 info_dict
= self
.getInfoDict()
257 self
.assertEqual(info_dict
, None)
259 def test_automatic_captions(self
):
260 self
.DL
.expect_warning(u
'Automatic Captions not supported by this server')
261 self
.DL
.params
['writeautomaticsub'] = True
262 self
.DL
.params
['subtitleslang'] = ['en']
263 subtitles
= self
.getSubtitles()
264 self
.assertTrue(len(subtitles
.keys()) == 0)
266 def test_nosubtitles(self
):
267 self
.DL
.expect_warning(u
'video doesn\'t have subtitles')
268 self
.url
= 'http://vimeo.com/56015672'
269 self
.DL
.params
['writesubtitles'] = True
270 self
.DL
.params
['allsubtitles'] = True
271 subtitles
= self
.getSubtitles()
272 self
.assertEqual(len(subtitles
), 0)
274 def test_multiple_langs(self
):
275 self
.DL
.params
['writesubtitles'] = True
276 langs
= ['es', 'fr', 'de']
277 self
.DL
.params
['subtitleslangs'] = langs
278 subtitles
= self
.getSubtitles()
280 self
.assertTrue(subtitles
.get(lang
) is not None, u
'Subtitles for \'%s\' not extracted' % lang
)
283 class TestWallaSubtitles(BaseTestSubtitles
):
284 url
= 'http://vod.walla.co.il/movie/2705958/the-yes-men'
287 def test_list_subtitles(self
):
288 self
.DL
.expect_warning(u
'Automatic Captions not supported by this server')
289 self
.DL
.params
['listsubtitles'] = True
290 info_dict
= self
.getInfoDict()
291 self
.assertEqual(info_dict
, None)
293 def test_allsubtitles(self
):
294 self
.DL
.expect_warning(u
'Automatic Captions not supported by this server')
295 self
.DL
.params
['writesubtitles'] = True
296 self
.DL
.params
['allsubtitles'] = True
297 subtitles
= self
.getSubtitles()
298 self
.assertEqual(set(subtitles
.keys()), set(['heb']))
299 self
.assertEqual(md5(subtitles
['heb']), 'e758c5d7cb982f6bef14f377ec7a3920')
301 def test_nosubtitles(self
):
302 self
.DL
.expect_warning(u
'video doesn\'t have subtitles')
303 self
.url
= 'http://vod.walla.co.il/movie/2642630/one-direction-all-for-one'
304 self
.DL
.params
['writesubtitles'] = True
305 self
.DL
.params
['allsubtitles'] = True
306 subtitles
= self
.getSubtitles()
307 self
.assertEqual(len(subtitles
), 0)
310 if __name__
== '__main__':