Merge pull request #2041 from dstftw/imdb-list

[youtube-dl] / test / test_playlists.py

diff --git a/test/test_playlists.py b/test/test_playlists.py
index 576f7fb4ea749f51be96f491ce645b3a95959901..9d522b3574c221bf6594cd9965ea17bf47b671c6 100644 (file)
--- a/test/test_playlists.py
+++ b/test/test_playlists.py
@@ -6,7 +6,6 @@
 import os
 import sys
 import unittest
-from youtube_dl.extractor.ivi import IviCompilationIE
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 from test.helper import FakeYDL
@@ -28,7 +27,9 @@ from youtube_dl.extractor import (
     BambuserChannelIE,
     BandcampAlbumIE,
     SmotriCommunityIE,
-    SmotriUserIE
+    SmotriUserIE,
+    IviCompilationIE,
+    ImdbListIE,
 )
 
 
@@ -187,6 +188,15 @@ class TestPlaylists(unittest.TestCase):
         self.assertEqual(result['id'], u'dezhurnyi_angel/season2')
         self.assertEqual(result['title'], u'Дежурный ангел (2010 - 2012) 2 сезон')
         self.assertTrue(len(result['entries']) >= 20)
+        
+    def test_imdb_list(self):
+        dl = FakeYDL()
+        ie = ImdbListIE(dl)
+        result = ie.extract('http://www.imdb.com/list/sMjedvGDd8U')
+        self.assertIsPlaylist(result)
+        self.assertEqual(result['id'], u'sMjedvGDd8U')
+        self.assertEqual(result['title'], u'Animated and Family Films')
+        self.assertTrue(len(result['entries']) >= 48)
 
 
 if __name__ == '__main__':