test/test_youtube_misc.py



    1 #!/usr/bin/env python
    2 from __future__ import unicode_literals
    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 
   10 
   11 from youtube_dl.extractor import YoutubeIE
   12 
   13 
   14 class TestYoutubeMisc(unittest.TestCase):
   15     def test_youtube_extract(self):
   16         assertExtractId = lambda url, id: self.assertEqual(YoutubeIE.extract_id(url), id)
   17         assertExtractId('http://www.youtube.com/watch?&v=BaW_jenozKc', 'BaW_jenozKc')
   18         assertExtractId('https://www.youtube.com/watch?&v=BaW_jenozKc', 'BaW_jenozKc')
   19         assertExtractId('https://www.youtube.com/watch?feature=player_embedded&v=BaW_jenozKc', 'BaW_jenozKc')
   20         assertExtractId('https://www.youtube.com/watch_popup?v=BaW_jenozKc', 'BaW_jenozKc')
   21         assertExtractId('http://www.youtube.com/watch?v=BaW_jenozKcsharePLED17F32AD9753930', 'BaW_jenozKc')
   22         assertExtractId('BaW_jenozKc', 'BaW_jenozKc')
   23 
   24 
   25 if __name__ == '__main__':
   26     unittest.main()