_ Git - youtube-dl/blob - test/test_ted_subtitles.py

   1 #!/usr/bin/env python
   2
   3 import sys
   4 import unittest
   5 import hashlib
   6
   7 # Allow direct execution
   8 import os
   9 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
  10
  11 from youtube_dl.extractor import TEDIE
  12 from youtube_dl.utils import *
  13 from helper import FakeYDL
  14
  15 md5 = lambda s: hashlib.md5(s.encode('utf-8')).hexdigest()
  16
  17 class TestTedSubtitles(unittest.TestCase):
  18     def setUp(self):
  19         self.DL = FakeYDL()
  20         self.url = 'http://www.ted.com/talks/dan_dennett_on_our_consciousness.html'
  21     def getInfoDict(self):
  22         IE = TEDIE(self.DL)
  23         info_dict = IE.extract(self.url)
  24         return info_dict
  25     def getSubtitles(self):
  26         info_dict = self.getInfoDict()
  27         return info_dict[0]['subtitles']
  28     def test_no_writesubtitles(self):
  29         subtitles = self.getSubtitles()
  30         self.assertEqual(subtitles, None)
  31     def test_subtitles(self):
  32         self.DL.params['writesubtitles'] = True
  33         subtitles = self.getSubtitles()
  34         self.assertEqual(md5(subtitles['en']), '2154f31ff9b9f89a0aa671537559c21d')
  35     def test_subtitles_lang(self):
  36         self.DL.params['writesubtitles'] = True
  37         self.DL.params['subtitleslangs'] = ['fr']
  38         subtitles = self.getSubtitles()
  39         self.assertEqual(md5(subtitles['fr']), '7616cbc6df20ec2c1204083c83871cf6')
  40     def test_allsubtitles(self):
  41         self.DL.params['writesubtitles'] = True
  42         self.DL.params['allsubtitles'] = True
  43         subtitles = self.getSubtitles()
  44         self.assertEqual(len(subtitles.keys()), 28)
  45     def test_list_subtitles(self):
  46         self.DL.params['listsubtitles'] = True
  47         info_dict = self.getInfoDict()
  48         self.assertEqual(info_dict, [None])
  49     def test_automatic_captions(self):
  50         self.DL.params['writeautomaticsub'] = True
  51         self.DL.params['subtitleslang'] = ['en']
  52         subtitles = self.getSubtitles()
  53         self.assertTrue(len(subtitles.keys()) == 0)
  54     def test_multiple_langs(self):
  55         self.DL.params['writesubtitles'] = True
  56         langs = ['es', 'fr', 'de']
  57         self.DL.params['subtitleslangs'] = langs
  58         subtitles = self.getSubtitles()
  59         for lang in langs:
  60             self.assertTrue(subtitles.get(lang) is not None, u'Subtitles for \'%s\' not extracted' % lang)
  61
  62 if __name__ == '__main__':
  63     unittest.main()