[yt-dlp.git] / test / test_subtitles.py

#!/usr/bin/env python
from __future__ import unicode_literals

# Allow direct execution
import os
import sys
import unittest
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from test.helper import FakeYDL, md5


from youtube_dl.extractor import (
    BlipTVIE,
    YoutubeIE,
    DailymotionIE,
    TEDIE,
    VimeoIE,
    WallaIE,
    CeskaTelevizeIE,
    LyndaIE,
    NPOIE,
    ComedyCentralIE,
    NRKTVIE,
)


class BaseTestSubtitles(unittest.TestCase):
    url = None
    IE = None

    def setUp(self):
        self.DL = FakeYDL()
        self.ie = self.IE()
        self.DL.add_info_extractor(self.ie)

    def getInfoDict(self):
        info_dict = self.DL.extract_info(self.url, download=False)
        return info_dict

    def getSubtitles(self):
        info_dict = self.getInfoDict()
        subtitles = info_dict['requested_subtitles']
        if not subtitles:
            return subtitles
        for sub_info in subtitles.values():
            if sub_info.get('data') is None:
                uf = self.DL.urlopen(sub_info['url'])
                sub_info['data'] = uf.read().decode('utf-8')
        return dict((l, sub_info['data']) for l, sub_info in subtitles.items())


class TestYoutubeSubtitles(BaseTestSubtitles):
    url = 'QRS8MkLhQmM'
    IE = YoutubeIE

    def test_youtube_subtitles(self):
        self.DL.params['writesubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['en']), '4cd9278a35ba2305f47354ee13472260')

    def test_youtube_subtitles_lang(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['subtitleslangs'] = ['it']
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['it']), '164a51f16f260476a05b50fe4c2f161d')

    def test_youtube_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(len(subtitles.keys()), 13)

    def test_youtube_subtitles_sbv_format(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['subtitlesformat'] = 'sbv'
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['en']), '13aeaa0c245a8bed9a451cb643e3ad8b')

    def test_youtube_subtitles_vtt_format(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['subtitlesformat'] = 'vtt'
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['en']), '3cb210999d3e021bd6c7f0ea751eab06')

    def test_youtube_list_subtitles(self):
        self.DL.expect_warning('Video doesn\'t have automatic captions')
        self.DL.params['listsubtitles'] = True
        info_dict = self.getInfoDict()
        self.assertEqual(info_dict, None)

    def test_youtube_automatic_captions(self):
        self.url = '8YoUxe5ncPo'
        self.DL.params['writeautomaticsub'] = True
        self.DL.params['subtitleslangs'] = ['it']
        subtitles = self.getSubtitles()
        self.assertTrue(subtitles['it'] is not None)

    def test_youtube_translated_subtitles(self):
        # This video has a subtitles track, which can be translated
        self.url = 'Ky9eprVWzlI'
        self.DL.params['writeautomaticsub'] = True
        self.DL.params['subtitleslangs'] = ['it']
        subtitles = self.getSubtitles()
        self.assertTrue(subtitles['it'] is not None)

    def test_youtube_nosubtitles(self):
        self.DL.expect_warning('video doesn\'t have subtitles')
        self.url = 'n5BB19UTcdA'
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(len(subtitles), 0)

    def test_youtube_multiple_langs(self):
        self.url = 'QRS8MkLhQmM'
        self.DL.params['writesubtitles'] = True
        langs = ['it', 'fr', 'de']
        self.DL.params['subtitleslangs'] = langs
        subtitles = self.getSubtitles()
        for lang in langs:
            self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)


class TestDailymotionSubtitles(BaseTestSubtitles):
    url = 'http://www.dailymotion.com/video/xczg00'
    IE = DailymotionIE

    def test_subtitles(self):
        self.DL.params['writesubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['en']), '976553874490cba125086bbfea3ff76f')

    def test_subtitles_lang(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['subtitleslangs'] = ['fr']
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['fr']), '594564ec7d588942e384e920e5341792')

    def test_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertTrue(len(subtitles.keys()) >= 6)

    def test_nosubtitles(self):
        self.DL.expect_warning('video doesn\'t have subtitles')
        self.url = 'http://www.dailymotion.com/video/x12u166_le-zapping-tele-star-du-08-aout-2013_tv'
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(len(subtitles), 0)

    def test_multiple_langs(self):
        self.DL.params['writesubtitles'] = True
        langs = ['es', 'fr', 'de']
        self.DL.params['subtitleslangs'] = langs
        subtitles = self.getSubtitles()
        for lang in langs:
            self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)


class TestTedSubtitles(BaseTestSubtitles):
    url = 'http://www.ted.com/talks/dan_dennett_on_our_consciousness.html'
    IE = TEDIE

    def test_no_writesubtitles(self):
        subtitles = self.getSubtitles()
        self.assertFalse(subtitles)

    def test_subtitles(self):
        self.DL.params['writesubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['en']), '4262c1665ff928a2dada178f62cb8d14')

    def test_subtitles_lang(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['subtitleslangs'] = ['fr']
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['fr']), '66a63f7f42c97a50f8c0e90bc7797bb5')

    def test_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertTrue(len(subtitles.keys()) >= 28)

    def test_list_subtitles(self):
        self.DL.params['listsubtitles'] = True
        info_dict = self.getInfoDict()
        self.assertEqual(info_dict, None)

    def test_multiple_langs(self):
        self.DL.params['writesubtitles'] = True
        langs = ['es', 'fr', 'de']
        self.DL.params['subtitleslangs'] = langs
        subtitles = self.getSubtitles()
        for lang in langs:
            self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)


class TestBlipTVSubtitles(BaseTestSubtitles):
    url = 'http://blip.tv/a/a-6603250'
    IE = BlipTVIE

    def test_list_subtitles(self):
        self.DL.params['listsubtitles'] = True
        info_dict = self.getInfoDict()
        self.assertEqual(info_dict, None)

    def test_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(set(subtitles.keys()), set(['en']))
        self.assertEqual(md5(subtitles['en']), '5b75c300af65fe4476dff79478bb93e4')


class TestVimeoSubtitles(BaseTestSubtitles):
    url = 'http://vimeo.com/76979871'
    IE = VimeoIE

    def test_subtitles(self):
        self.DL.params['writesubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['en']), '8062383cf4dec168fc40a088aa6d5888')

    def test_subtitles_lang(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['subtitleslangs'] = ['fr']
        subtitles = self.getSubtitles()
        self.assertEqual(md5(subtitles['fr']), 'b6191146a6c5d3a452244d853fde6dc8')

    def test_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(set(subtitles.keys()), set(['de', 'en', 'es', 'fr']))

    def test_nosubtitles(self):
        self.DL.expect_warning('video doesn\'t have subtitles')
        self.url = 'http://vimeo.com/56015672'
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(len(subtitles), 0)

    def test_multiple_langs(self):
        self.DL.params['writesubtitles'] = True
        langs = ['es', 'fr', 'de']
        self.DL.params['subtitleslangs'] = langs
        subtitles = self.getSubtitles()
        for lang in langs:
            self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)


class TestWallaSubtitles(BaseTestSubtitles):
    url = 'http://vod.walla.co.il/movie/2705958/the-yes-men'
    IE = WallaIE

    def test_list_subtitles(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['listsubtitles'] = True
        info_dict = self.getInfoDict()
        self.assertEqual(info_dict, None)

    def test_allsubtitles(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(set(subtitles.keys()), set(['heb']))
        self.assertEqual(md5(subtitles['heb']), 'e758c5d7cb982f6bef14f377ec7a3920')

    def test_nosubtitles(self):
        self.DL.expect_warning('video doesn\'t have subtitles')
        self.url = 'http://vod.walla.co.il/movie/2642630/one-direction-all-for-one'
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(len(subtitles), 0)


class TestCeskaTelevizeSubtitles(BaseTestSubtitles):
    url = 'http://www.ceskatelevize.cz/ivysilani/10600540290-u6-uzasny-svet-techniky'
    IE = CeskaTelevizeIE

    def test_list_subtitles(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['listsubtitles'] = True
        info_dict = self.getInfoDict()
        self.assertEqual(info_dict, None)

    def test_allsubtitles(self):
        self.DL.expect_warning('Automatic Captions not supported by this server')
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(set(subtitles.keys()), set(['cs']))
        self.assertTrue(len(subtitles['cs']) > 20000)

    def test_nosubtitles(self):
        self.DL.expect_warning('video doesn\'t have subtitles')
        self.url = 'http://www.ceskatelevize.cz/ivysilani/ivysilani/10441294653-hyde-park-civilizace/214411058091220'
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(len(subtitles), 0)


class TestLyndaSubtitles(BaseTestSubtitles):
    url = 'http://www.lynda.com/Bootstrap-tutorials/Using-exercise-files/110885/114408-4.html'
    IE = LyndaIE

    def test_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(set(subtitles.keys()), set(['en']))
        self.assertEqual(md5(subtitles['en']), '09bbe67222259bed60deaa26997d73a7')


class TestNPOSubtitles(BaseTestSubtitles):
    url = 'http://www.npo.nl/nos-journaal/28-08-2014/POW_00722860'
    IE = NPOIE

    def test_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(set(subtitles.keys()), set(['nl']))
        self.assertEqual(md5(subtitles['nl']), 'fc6435027572b63fb4ab143abd5ad3f4')


class TestMTVSubtitles(BaseTestSubtitles):
    url = 'http://www.cc.com/video-clips/kllhuv/stand-up-greg-fitzsimmons--uncensored---too-good-of-a-mother'
    IE = ComedyCentralIE

    def getInfoDict(self):
        return super(TestMTVSubtitles, self).getInfoDict()['entries'][0]

    def test_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(set(subtitles.keys()), set(['en']))
        self.assertEqual(md5(subtitles['en']), 'b9f6ca22a6acf597ec76f61749765e65')


class TestNRKSubtitles(BaseTestSubtitles):
    url = 'http://tv.nrk.no/serie/ikke-gjoer-dette-hjemme/DMPV73000411/sesong-2/episode-1'
    IE = NRKTVIE

    def test_allsubtitles(self):
        self.DL.params['writesubtitles'] = True
        self.DL.params['allsubtitles'] = True
        subtitles = self.getSubtitles()
        self.assertEqual(set(subtitles.keys()), set(['no']))
        self.assertEqual(md5(subtitles['no']), '1d221e6458c95c5494dcd38e6a1f129a')


if __name__ == '__main__':
    unittest.main()
Commit	Line	Data
ae84f879	1	#!/usr/bin/env python
a0f59cdc	2	from __future__ import unicode_literals
ae84f879 JMF	3
	4	# Allow direct execution
	5	import os
	6	import sys
	7	import unittest
	8	sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
	9
d0efb9ec	10	from test.helper import FakeYDL, md5
ae84f879 JMF	11
	12
	13	from youtube_dl.extractor import (
b4bcffef	14	BlipTVIE,
ae84f879 JMF	15	YoutubeIE,
	16	DailymotionIE,
	17	TEDIE,
4edff4cf	18	VimeoIE,
7bc8780c	19	WallaIE,
0b54a5b1	20	CeskaTelevizeIE,
311c3938	21	LyndaIE,
b9b42f2e	22	NPOIE,
0af25f78	23	ComedyCentralIE,
01561da1	24	NRKTVIE,
ae84f879 JMF	25	)
	26
	27
	28	class BaseTestSubtitles(unittest.TestCase):
	29	url = None
	30	IE = None
5f6a1245	31
ae84f879 JMF	32	def setUp(self):
ae84f879 JMF	33	self.DL = FakeYDL()
a504ced0 JMF	34	self.ie = self.IE()
a504ced0 JMF	35	self.DL.add_info_extractor(self.ie)
ae84f879 JMF	36
ae84f879 JMF	37	def getInfoDict(self):
a504ced0	38	info_dict = self.DL.extract_info(self.url, download=False)
ae84f879 JMF	39	return info_dict
	40
	41	def getSubtitles(self):
	42	info_dict = self.getInfoDict()
c84dd8a9	43	subtitles = info_dict['requested_subtitles']
a504ced0 JMF	44	if not subtitles:
	45	return subtitles
	46	for sub_info in subtitles.values():
	47	if sub_info.get('data') is None:
	48	uf = self.DL.urlopen(sub_info['url'])
	49	sub_info['data'] = uf.read().decode('utf-8')
	50	return dict((l, sub_info['data']) for l, sub_info in subtitles.items())
ae84f879 JMF	51
	52
	53	class TestYoutubeSubtitles(BaseTestSubtitles):
	54	url = 'QRS8MkLhQmM'
	55	IE = YoutubeIE
	56
ae84f879 JMF	57	def test_youtube_subtitles(self):
	58	self.DL.params['writesubtitles'] = True
	59	subtitles = self.getSubtitles()
	60	self.assertEqual(md5(subtitles['en']), '4cd9278a35ba2305f47354ee13472260')
	61
	62	def test_youtube_subtitles_lang(self):
	63	self.DL.params['writesubtitles'] = True
	64	self.DL.params['subtitleslangs'] = ['it']
	65	subtitles = self.getSubtitles()
	66	self.assertEqual(md5(subtitles['it']), '164a51f16f260476a05b50fe4c2f161d')
	67
	68	def test_youtube_allsubtitles(self):
	69	self.DL.params['writesubtitles'] = True
	70	self.DL.params['allsubtitles'] = True
	71	subtitles = self.getSubtitles()
	72	self.assertEqual(len(subtitles.keys()), 13)
	73
	74	def test_youtube_subtitles_sbv_format(self):
	75	self.DL.params['writesubtitles'] = True
	76	self.DL.params['subtitlesformat'] = 'sbv'
	77	subtitles = self.getSubtitles()
	78	self.assertEqual(md5(subtitles['en']), '13aeaa0c245a8bed9a451cb643e3ad8b')
	79
	80	def test_youtube_subtitles_vtt_format(self):
	81	self.DL.params['writesubtitles'] = True
	82	self.DL.params['subtitlesformat'] = 'vtt'
	83	subtitles = self.getSubtitles()
4e0084d9	84	self.assertEqual(md5(subtitles['en']), '3cb210999d3e021bd6c7f0ea751eab06')
ae84f879 JMF	85
ae84f879 JMF	86	def test_youtube_list_subtitles(self):
a0f59cdc	87	self.DL.expect_warning('Video doesn\'t have automatic captions')
ae84f879 JMF	88	self.DL.params['listsubtitles'] = True
	89	info_dict = self.getInfoDict()
	90	self.assertEqual(info_dict, None)
	91
	92	def test_youtube_automatic_captions(self):
	93	self.url = '8YoUxe5ncPo'
	94	self.DL.params['writeautomaticsub'] = True
	95	self.DL.params['subtitleslangs'] = ['it']
	96	subtitles = self.getSubtitles()
	97	self.assertTrue(subtitles['it'] is not None)
	98
7d900ef1 JMF	99	def test_youtube_translated_subtitles(self):
	100	# This video has a subtitles track, which can be translated
	101	self.url = 'Ky9eprVWzlI'
	102	self.DL.params['writeautomaticsub'] = True
	103	self.DL.params['subtitleslangs'] = ['it']
	104	subtitles = self.getSubtitles()
	105	self.assertTrue(subtitles['it'] is not None)
	106
ae84f879	107	def test_youtube_nosubtitles(self):
a0f59cdc	108	self.DL.expect_warning('video doesn\'t have subtitles')
345e3783	109	self.url = 'n5BB19UTcdA'
ae84f879 JMF	110	self.DL.params['writesubtitles'] = True
	111	self.DL.params['allsubtitles'] = True
	112	subtitles = self.getSubtitles()
	113	self.assertEqual(len(subtitles), 0)
	114
	115	def test_youtube_multiple_langs(self):
	116	self.url = 'QRS8MkLhQmM'
	117	self.DL.params['writesubtitles'] = True
	118	langs = ['it', 'fr', 'de']
	119	self.DL.params['subtitleslangs'] = langs
	120	subtitles = self.getSubtitles()
	121	for lang in langs:
a0f59cdc	122	self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)
ae84f879 JMF	123
	124
	125	class TestDailymotionSubtitles(BaseTestSubtitles):
	126	url = 'http://www.dailymotion.com/video/xczg00'
	127	IE = DailymotionIE
	128
ae84f879 JMF	129	def test_subtitles(self):
	130	self.DL.params['writesubtitles'] = True
	131	subtitles = self.getSubtitles()
	132	self.assertEqual(md5(subtitles['en']), '976553874490cba125086bbfea3ff76f')
	133
	134	def test_subtitles_lang(self):
	135	self.DL.params['writesubtitles'] = True
	136	self.DL.params['subtitleslangs'] = ['fr']
	137	subtitles = self.getSubtitles()
	138	self.assertEqual(md5(subtitles['fr']), '594564ec7d588942e384e920e5341792')
	139
	140	def test_allsubtitles(self):
	141	self.DL.params['writesubtitles'] = True
	142	self.DL.params['allsubtitles'] = True
	143	subtitles = self.getSubtitles()
8fb474fb	144	self.assertTrue(len(subtitles.keys()) >= 6)
ae84f879	145
ae84f879	146	def test_nosubtitles(self):
a0f59cdc	147	self.DL.expect_warning('video doesn\'t have subtitles')
ae84f879 JMF	148	self.url = 'http://www.dailymotion.com/video/x12u166_le-zapping-tele-star-du-08-aout-2013_tv'
	149	self.DL.params['writesubtitles'] = True
	150	self.DL.params['allsubtitles'] = True
	151	subtitles = self.getSubtitles()
	152	self.assertEqual(len(subtitles), 0)
	153
	154	def test_multiple_langs(self):
	155	self.DL.params['writesubtitles'] = True
	156	langs = ['es', 'fr', 'de']
	157	self.DL.params['subtitleslangs'] = langs
	158	subtitles = self.getSubtitles()
	159	for lang in langs:
a0f59cdc	160	self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)
ae84f879 JMF	161
	162
	163	class TestTedSubtitles(BaseTestSubtitles):
	164	url = 'http://www.ted.com/talks/dan_dennett_on_our_consciousness.html'
	165	IE = TEDIE
	166
	167	def test_no_writesubtitles(self):
	168	subtitles = self.getSubtitles()
a504ced0	169	self.assertFalse(subtitles)
ae84f879 JMF	170
	171	def test_subtitles(self):
	172	self.DL.params['writesubtitles'] = True
	173	subtitles = self.getSubtitles()
26dca166	174	self.assertEqual(md5(subtitles['en']), '4262c1665ff928a2dada178f62cb8d14')
ae84f879 JMF	175
	176	def test_subtitles_lang(self):
	177	self.DL.params['writesubtitles'] = True
	178	self.DL.params['subtitleslangs'] = ['fr']
	179	subtitles = self.getSubtitles()
26dca166	180	self.assertEqual(md5(subtitles['fr']), '66a63f7f42c97a50f8c0e90bc7797bb5')
ae84f879 JMF	181
	182	def test_allsubtitles(self):
	183	self.DL.params['writesubtitles'] = True
	184	self.DL.params['allsubtitles'] = True
	185	subtitles = self.getSubtitles()
0321213c	186	self.assertTrue(len(subtitles.keys()) >= 28)
ae84f879 JMF	187
ae84f879 JMF	188	def test_list_subtitles(self):
ae84f879 JMF	189	self.DL.params['listsubtitles'] = True
	190	info_dict = self.getInfoDict()
	191	self.assertEqual(info_dict, None)
	192
ae84f879 JMF	193	def test_multiple_langs(self):
	194	self.DL.params['writesubtitles'] = True
	195	langs = ['es', 'fr', 'de']
	196	self.DL.params['subtitleslangs'] = langs
	197	subtitles = self.getSubtitles()
	198	for lang in langs:
a0f59cdc	199	self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)
ae84f879	200
b4bcffef PH	201
	202	class TestBlipTVSubtitles(BaseTestSubtitles):
	203	url = 'http://blip.tv/a/a-6603250'
	204	IE = BlipTVIE
	205
	206	def test_list_subtitles(self):
b4bcffef PH	207	self.DL.params['listsubtitles'] = True
	208	info_dict = self.getInfoDict()
	209	self.assertEqual(info_dict, None)
	210
	211	def test_allsubtitles(self):
b4bcffef PH	212	self.DL.params['writesubtitles'] = True
	213	self.DL.params['allsubtitles'] = True
	214	subtitles = self.getSubtitles()
	215	self.assertEqual(set(subtitles.keys()), set(['en']))
	216	self.assertEqual(md5(subtitles['en']), '5b75c300af65fe4476dff79478bb93e4')
	217
	218
4edff4cf S	219	class TestVimeoSubtitles(BaseTestSubtitles):
	220	url = 'http://vimeo.com/76979871'
	221	IE = VimeoIE
b0268cb6	222
4edff4cf S	223	def test_subtitles(self):
	224	self.DL.params['writesubtitles'] = True
	225	subtitles = self.getSubtitles()
36e7a4ca	226	self.assertEqual(md5(subtitles['en']), '8062383cf4dec168fc40a088aa6d5888')
4edff4cf S	227
	228	def test_subtitles_lang(self):
	229	self.DL.params['writesubtitles'] = True
	230	self.DL.params['subtitleslangs'] = ['fr']
	231	subtitles = self.getSubtitles()
	232	self.assertEqual(md5(subtitles['fr']), 'b6191146a6c5d3a452244d853fde6dc8')
	233
	234	def test_allsubtitles(self):
	235	self.DL.params['writesubtitles'] = True
	236	self.DL.params['allsubtitles'] = True
	237	subtitles = self.getSubtitles()
	238	self.assertEqual(set(subtitles.keys()), set(['de', 'en', 'es', 'fr']))
	239
4edff4cf	240	def test_nosubtitles(self):
a0f59cdc	241	self.DL.expect_warning('video doesn\'t have subtitles')
4edff4cf S	242	self.url = 'http://vimeo.com/56015672'
	243	self.DL.params['writesubtitles'] = True
	244	self.DL.params['allsubtitles'] = True
	245	subtitles = self.getSubtitles()
	246	self.assertEqual(len(subtitles), 0)
	247
	248	def test_multiple_langs(self):
	249	self.DL.params['writesubtitles'] = True
	250	langs = ['es', 'fr', 'de']
	251	self.DL.params['subtitleslangs'] = langs
	252	subtitles = self.getSubtitles()
	253	for lang in langs:
a0f59cdc	254	self.assertTrue(subtitles.get(lang) is not None, 'Subtitles for \'%s\' not extracted' % lang)
4edff4cf S	255
4edff4cf S	256
0ca41c3d	257	class TestWallaSubtitles(BaseTestSubtitles):
7bc8780c S	258	url = 'http://vod.walla.co.il/movie/2705958/the-yes-men'
	259	IE = WallaIE
	260
	261	def test_list_subtitles(self):
a0f59cdc	262	self.DL.expect_warning('Automatic Captions not supported by this server')
7bc8780c S	263	self.DL.params['listsubtitles'] = True
	264	info_dict = self.getInfoDict()
	265	self.assertEqual(info_dict, None)
	266
	267	def test_allsubtitles(self):
a0f59cdc	268	self.DL.expect_warning('Automatic Captions not supported by this server')
7bc8780c S	269	self.DL.params['writesubtitles'] = True
	270	self.DL.params['allsubtitles'] = True
	271	subtitles = self.getSubtitles()
	272	self.assertEqual(set(subtitles.keys()), set(['heb']))
	273	self.assertEqual(md5(subtitles['heb']), 'e758c5d7cb982f6bef14f377ec7a3920')
	274
	275	def test_nosubtitles(self):
a0f59cdc	276	self.DL.expect_warning('video doesn\'t have subtitles')
7bc8780c S	277	self.url = 'http://vod.walla.co.il/movie/2642630/one-direction-all-for-one'
	278	self.DL.params['writesubtitles'] = True
	279	self.DL.params['allsubtitles'] = True
	280	subtitles = self.getSubtitles()
	281	self.assertEqual(len(subtitles), 0)
	282
	283
0b54a5b1 S	284	class TestCeskaTelevizeSubtitles(BaseTestSubtitles):
	285	url = 'http://www.ceskatelevize.cz/ivysilani/10600540290-u6-uzasny-svet-techniky'
	286	IE = CeskaTelevizeIE
	287
	288	def test_list_subtitles(self):
	289	self.DL.expect_warning('Automatic Captions not supported by this server')
	290	self.DL.params['listsubtitles'] = True
	291	info_dict = self.getInfoDict()
	292	self.assertEqual(info_dict, None)
	293
	294	def test_allsubtitles(self):
	295	self.DL.expect_warning('Automatic Captions not supported by this server')
	296	self.DL.params['writesubtitles'] = True
	297	self.DL.params['allsubtitles'] = True
	298	subtitles = self.getSubtitles()
	299	self.assertEqual(set(subtitles.keys()), set(['cs']))
8fb474fb	300	self.assertTrue(len(subtitles['cs']) > 20000)
0b54a5b1 S	301
	302	def test_nosubtitles(self):
	303	self.DL.expect_warning('video doesn\'t have subtitles')
	304	self.url = 'http://www.ceskatelevize.cz/ivysilani/ivysilani/10441294653-hyde-park-civilizace/214411058091220'
	305	self.DL.params['writesubtitles'] = True
	306	self.DL.params['allsubtitles'] = True
	307	subtitles = self.getSubtitles()
	308	self.assertEqual(len(subtitles), 0)
	309
	310
311c3938 JMF	311	class TestLyndaSubtitles(BaseTestSubtitles):
	312	url = 'http://www.lynda.com/Bootstrap-tutorials/Using-exercise-files/110885/114408-4.html'
	313	IE = LyndaIE
	314
	315	def test_allsubtitles(self):
	316	self.DL.params['writesubtitles'] = True
	317	self.DL.params['allsubtitles'] = True
	318	subtitles = self.getSubtitles()
	319	self.assertEqual(set(subtitles.keys()), set(['en']))
	320	self.assertEqual(md5(subtitles['en']), '09bbe67222259bed60deaa26997d73a7')
	321
	322
b9b42f2e JMF	323	class TestNPOSubtitles(BaseTestSubtitles):
	324	url = 'http://www.npo.nl/nos-journaal/28-08-2014/POW_00722860'
	325	IE = NPOIE
	326
	327	def test_allsubtitles(self):
	328	self.DL.params['writesubtitles'] = True
	329	self.DL.params['allsubtitles'] = True
	330	subtitles = self.getSubtitles()
	331	self.assertEqual(set(subtitles.keys()), set(['nl']))
	332	self.assertEqual(md5(subtitles['nl']), 'fc6435027572b63fb4ab143abd5ad3f4')
	333
	334
0af25f78 JMF	335	class TestMTVSubtitles(BaseTestSubtitles):
	336	url = 'http://www.cc.com/video-clips/kllhuv/stand-up-greg-fitzsimmons--uncensored---too-good-of-a-mother'
	337	IE = ComedyCentralIE
	338
	339	def getInfoDict(self):
	340	return super(TestMTVSubtitles, self).getInfoDict()['entries'][0]
	341
	342	def test_allsubtitles(self):
	343	self.DL.params['writesubtitles'] = True
	344	self.DL.params['allsubtitles'] = True
	345	subtitles = self.getSubtitles()
	346	self.assertEqual(set(subtitles.keys()), set(['en']))
	347	self.assertEqual(md5(subtitles['en']), 'b9f6ca22a6acf597ec76f61749765e65')
	348
	349
01561da1 JMF	350	class TestNRKSubtitles(BaseTestSubtitles):
	351	url = 'http://tv.nrk.no/serie/ikke-gjoer-dette-hjemme/DMPV73000411/sesong-2/episode-1'
	352	IE = NRKTVIE
	353
	354	def test_allsubtitles(self):
	355	self.DL.params['writesubtitles'] = True
	356	self.DL.params['allsubtitles'] = True
	357	subtitles = self.getSubtitles()
	358	self.assertEqual(set(subtitles.keys()), set(['no']))
	359	self.assertEqual(md5(subtitles['no']), '1d221e6458c95c5494dcd38e6a1f129a')
	360
	361
ae84f879 JMF	362	if __name__ == '__main__':
ae84f879 JMF	363	unittest.main()