[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,
    ThePlatformIE,
)


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.assertFalse(subtitles)

    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.assertFalse(subtitles)

    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.assertFalse(subtitles)

    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.assertFalse(subtitles)


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.assertFalse(subtitles)


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


class TestThePlatformSubtitles(BaseTestSubtitles):
    # from http://www.3playmedia.com/services-features/tools/integrations/theplatform/
    # (see http://theplatform.com/about/partners/type/subtitles-closed-captioning/)
    url = 'theplatform:JFUjUE1_ehvq'
    IE = ThePlatformIE

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


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