test_youtube_misc.py (1006B)
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()