1
0
mirror of https://github.com/ytdl-org/youtube-dl.git synced 2025-01-27 13:19:53 +00:00

Add PBS extractor subtitle tests

This commit is contained in:
Raphaël Doré 2018-09-04 14:39:31 -04:00
parent 6b4130962c
commit 81474531ff

View File

@ -28,6 +28,7 @@ from youtube_dl.extractor import (
RTVEALaCartaIE,
FunnyOrDieIE,
DemocracynowIE,
PBSIE
)
@ -354,5 +355,36 @@ class TestDemocracynowSubtitles(BaseTestSubtitles):
self.assertEqual(md5(subtitles['en']), 'acaca989e24a9e45a6719c9b3d60815c')
class TestPBSSubtitles(BaseTestSubtitles):
url = 'https://www.pbs.org/video/how-fantasy-reflects-our-world-picecq/'
IE = PBSIE
def test_allsubtitles(self):
self.DL.params['writesubtitles'] = True
self.DL.params['allsubtitles'] = True
subtitles = self.getSubtitles()
self.assertEqual(len(subtitles.keys()), 1)
self.assertIn('en', subtitles)
self.assertTrue(len(subtitles['en']) > 20000)
def test_subtitles_dfxp_format(self):
self.DL.params['writesubtitles'] = True
self.DL.params['subtitlesformat'] = 'dfxp'
subtitles = self.getSubtitles()
self.assertEqual(md5(subtitles['en']), '643b034254cdc3768ff1e750b6b5873b')
def test_subtitles_vtt_format(self):
self.DL.params['writesubtitles'] = True
self.DL.params['subtitlesformat'] = 'vtt'
subtitles = self.getSubtitles()
self.assertEqual(md5(subtitles['en']), '937a05711555b165d4c55a9667017045')
def test_subtitles_srt_format(self):
self.DL.params['writesubtitles'] = True
self.DL.params['subtitlesformat'] = 'srt'
subtitles = self.getSubtitles()
self.assertEqual(md5(subtitles['en']), '2082c21b43759d9bf172931b2f2ca371')
if __name__ == '__main__':
unittest.main()