[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,
    RaiIE,
    VikiIE,
)


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')


class TestRaiSubtitles(BaseTestSubtitles):
    url = 'http://www.rai.tv/dl/RaiTV/programmi/media/ContentItem-cb27157f-9dd0-4aee-b788-b1f67643a391.html'
    IE = RaiIE

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


class TestVikiSubtitles(BaseTestSubtitles):
    url = 'http://www.viki.com/videos/1060846v-punch-episode-18'
    IE = VikiIE

    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']), 'b0b781eeb45efd3f6398a925b259150b')


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