travisCI mod

pytube/contrib/playlist.py

@@ -176,14 +176,20 @@ class Playlist(object):
 
     def title(self):
         """return playlist title (name)
-        """        
+        """
         try:
             url = self.construct_playlist_url()
-            req = request.get(url)           
+            req = request.get(url)
             open_tag = "<title>"
             end_tag = "</title>"
-            matchresult = re.compile(open_tag + "(.+?)" + end_tag).search(req).group().replace(open_tag,"").replace(end_tag,"").replace("- YouTube","").strip()
+            matchresult = re.compile(open_tag + "(.+?)" + end_tag)
+            matchresult = matchresult.search(req).group()
+            matchresult = matchresult.replace(open_tag,"")
+            matchresult = matchresult.replace(end_tag,"")
+            matchresult = matchresult.replace("- YouTube","")
+            matchresult = matchresult.strip()
+
             return matchresult
         except Exception as e:
             logger.debug(e)
-            return None
+            return None

tests/contrib/test_playlist.py

@@ -1,6 +1,8 @@
-from pytube import YouTube, Playlist
+from pytube import Playlist
+
 
 def test_title():
-	pl = Playlist("https://www.youtube.com/watch?v=QXeEoD0pB3E&list=PLsyeobzWxl7poL9JTVyndKe62ieoN-MZ3")
-	pl_title = pl.title()
-	assert pl_title == "Python Tutorial for Beginners"
+    url = "https://www.youtube.com/watch?v=QXeEoD0pB3E&list=PLsyeobzWxl7poL9JTVyndKe62ieoN-MZ3"
+    pl = Playlist(url)
+    pl_title = pl.title()
+    assert pl_title == "Python Tutorial for Beginners"