don't download real youtube data when testing playlist

pytube/contrib/playlist.py

@@ -45,10 +45,9 @@ class Playlist:
         and returns the "load more" url if found.
         """
         match = re.search(
-                r"data-uix-load-more-href=\"(/browse_ajax\?"
-                'action_continuation=.*?)"',
-                req,
-            )
+            r"data-uix-load-more-href=\"(/browse_ajax\?" 'action_continuation=.*?)"',
+            req,
+        )
         if match:
             return "https://www.youtube.com" + match.group(1)
 
@@ -79,7 +78,9 @@ class Playlist:
             )
             # remove duplicates
             link_list.extend(list(OrderedDict.fromkeys(videos)))
-            load_more_url = self._find_load_more_url(load_more["load_more_widget_html"], )
+            load_more_url = self._find_load_more_url(
+                load_more["load_more_widget_html"],
+            )
 
         return link_list
 
@@ -120,7 +121,7 @@ class Playlist:
         download_path: Optional[str] = None,
         prefix_number: bool = True,
         reverse_numbering: bool = False,
-        resolution:str = "720p"
+        resolution: str = "720p",
     ) -> None:
         """Download all the videos in the the playlist. Initially, download
         resolution is 720p (or highest available), later more option
@@ -158,7 +159,10 @@ class Playlist:
                 if not self.suppress_exception:
                     raise e
             else:
-                dl_stream = yt.streams.get_by_resolution(resolution=resolution) or yt.streams.get_lowest_resolution()
+                dl_stream = (
+                    yt.streams.get_by_resolution(resolution=resolution)
+                    or yt.streams.get_lowest_resolution()
+                )
                 assert dl_stream is not None
 
                 logger.debug("download path: %s", download_path)

pytube/helpers.py

@@ -102,9 +102,8 @@ def create_logger(level: int = logging.ERROR) -> logging.Logger:
     return logger
 
 
-T = TypeVar('T')
+T = TypeVar("T")
 
 
 def cache(func: Callable[..., T]) -> T:
     return functools.lru_cache()(func)  # type: ignore
-

pytube/itags.py

@@ -118,5 +118,5 @@ def get_format_profile(itag: int) -> Dict:
         "is_3d": itag in _3D,
         "is_hdr": itag in HDR,
         "fps": 60 if itag in _60FPS else 30,
-        "is_dash": itag in DASH_MP4_VIDEO or itag in DASH_MP4_AUDIO
+        "is_dash": itag in DASH_MP4_VIDEO or itag in DASH_MP4_AUDIO,
     }

pytube/query.py

@@ -221,7 +221,7 @@ class StreamQuery:
         """
         return self
 
-    def get_by_itag(self, itag:int) -> Optional[Stream]:
+    def get_by_itag(self, itag: int) -> Optional[Stream]:
         """Get the corresponding :class:`Stream <Stream>` for a given itag.
 
         :param int itag:
@@ -234,7 +234,7 @@ class StreamQuery:
         """
         return self.itag_index.get(int(itag))
 
-    def get_by_resolution(self, resolution:str) -> Optional[Stream]:
+    def get_by_resolution(self, resolution: str) -> Optional[Stream]:
         """Get the corresponding :class:`Stream <Stream>` for a given resolution.
         Stream must be a progressive mp4.
 
@@ -247,7 +247,7 @@ class StreamQuery:
 
         """
         return self.filter(
-            progressive=True, subtype='mp4', resolution=resolution
+            progressive=True, subtype="mp4", resolution=resolution
         ).first()
 
     def get_lowest_resolution(self) -> Optional[Stream]:
@@ -259,7 +259,12 @@ class StreamQuery:
             not found.
 
         """
-        return self.filter(progressive=True, subtype='mp4').order_by('resolution').desc().last()
+        return (
+            self.filter(progressive=True, subtype="mp4")
+            .order_by("resolution")
+            .desc()
+            .last()
+        )
 
     def first(self) -> Optional[Stream]:
         """Get the first :class:`Stream <Stream>` in the results.

pytube/request.py

@@ -5,7 +5,7 @@ from urllib.request import Request
 from urllib.request import urlopen
 
 
-def _execute_request(url:str) -> Any:
+def _execute_request(url: str) -> Any:
     return urlopen(Request(url, headers={"User-Agent": "Mozilla/5.0"}))
 
 
@@ -21,7 +21,7 @@ def get(url) -> str:
     return _execute_request(url).read().decode("utf-8")
 
 
-def stream(url: str, chunk_size:int = 8192) -> Iterable[bytes]:
+def stream(url: str, chunk_size: int = 8192) -> Iterable[bytes]:
     """Read the response in chunks.
     :param str url:
         The URL to perform the GET request for.

tests/contrib/test_playlist.py

@@ -1,10 +1,13 @@
 # -*- coding: utf-8 -*-
+from unittest.mock import MagicMock
+
 from pytube import Playlist
 
 
+
 def test_title():
     list_key = "PLsyeobzWxl7poL9JTVyndKe62ieoN-MZ3"
-    url = "https://www.youtube.com/playlist?list=" + list_key
+    url = "https://www.fakeurl.com/playlist?list=" + list_key
     pl = Playlist(url)
     pl_title = pl.title()
     assert pl_title == "Python Tutorial for Beginners"