test reprs

pytube/contrib/playlist.py

@@ -155,7 +155,7 @@ class Playlist(Sequence):
     @deprecated(
         "This call is unnecessary, you can directly access .video_urls or .videos"
     )
-    def populate_video_urls(self) -> List[str]:
+    def populate_video_urls(self) -> List[str]:  # pragma: no cover
         """Complete links of all the videos in playlist
 
         :rtype: List[str]

pytube/query.py

@@ -327,16 +327,18 @@ class StreamQuery(Sequence):
             pass
 
     @deprecated("Get the size of this list directly using len()")
-    def count(self) -> int:
-        """Get the count the query would return.
+    def count(self, value: Optional[str] = None) -> int:  # pragma: no cover
+        """Get the count of items in the list.
 
         :rtype: int
-
         """
-        return len(self)
+        if value:
+            return self.fmt_streams.count(value)
+        else:
+            return len(self)
 
     @deprecated("This object can be treated as a list, all() is useless")
-    def all(self) -> List[Stream]:
+    def all(self) -> List[Stream]:  # pragma: no cover
         """Get all the results represented by this query as a list.
 
         :rtype: list
@@ -380,7 +382,7 @@ class CaptionQuery(Sequence):
         return self.lang_code_index.get(lang_code)
 
     @deprecated("This object can be treated as a list, all() is useless")
-    def all(self) -> List[Caption]:
+    def all(self) -> List[Caption]:  # pragma: no cover
         """Get all the results represented by this query as a list.
 
         :rtype: list

tests/contrib/test_playlist.py

@@ -81,6 +81,29 @@ def test_video_urls(request_get, playlist_html):
     ]
 
 
+@mock.patch("pytube.contrib.playlist.request.get")
+def test_repr(request_get, playlist_html):
+    url = "https://www.fakeurl.com/playlist?list=whatever"
+    request_get.return_value = playlist_html
+    playlist = Playlist(url)
+    playlist._find_load_more_url = MagicMock(return_value=None)
+    request_get.assert_called()
+    assert (
+        repr(playlist) == "['https://www.youtube.com/watch?v=ujTCoH21GlA', "
+        "'https://www.youtube.com/watch?v=45ryDIPHdGg', "
+        "'https://www.youtube.com/watch?v=1BYu65vLKdA', "
+        "'https://www.youtube.com/watch?v=3AQ_74xrch8', "
+        "'https://www.youtube.com/watch?v=ddqQUz9mZaM', "
+        "'https://www.youtube.com/watch?v=vwLT6bZrHEE', "
+        "'https://www.youtube.com/watch?v=TQKI0KE-JYY', "
+        "'https://www.youtube.com/watch?v=dNBvQ38MlT8', "
+        "'https://www.youtube.com/watch?v=JHxyrMgOUWI', "
+        "'https://www.youtube.com/watch?v=l2I8NycJMCY', "
+        "'https://www.youtube.com/watch?v=g1Zbuk1gAfk', "
+        "'https://www.youtube.com/watch?v=zixd-si9Q-o']"
+    )
+
+
 @mock.patch("pytube.contrib.playlist.request.get")
 def test_sequence(request_get, playlist_html):
     url = "https://www.fakeurl.com/playlist?list=whatever"

tests/test_captions.py

@@ -103,6 +103,9 @@ def test_repr():
     )
     assert str(caption) == '<Caption lang="name1" code="en">'
 
+    caption_query = CaptionQuery(captions=[caption])
+    assert repr(caption_query) == '[<Caption lang="name1" code="en">]'
+
 
 @mock.patch("pytube.request.get")
 def test_xml_captions(request_get):

tests/test_mixins.py

@@ -1,3 +0,0 @@
-# -*- coding: utf-8 -*-
-def test_pre_signed_video(presigned_video):
-    assert presigned_video.streams.count() == 12

tests/test_query.py

@@ -159,3 +159,15 @@ def test_otf(cipher_signature):
 
     otf = cipher_signature.streams.otf(True).all()
     assert len(otf) == 0
+
+
+def test_repr(cipher_signature):
+    assert repr(
+        cipher_signature.streams.filter(
+            progressive=True, subtype="mp4", resolution="360p"
+        )
+    ) == (
+        '[<Stream: itag="18" mime_type="video/mp4" '
+        'res="360p" fps="30fps" vcodec="avc1.42001E" '
+        'acodec="mp4a.40.2" progressive="True" type="video">]'
+    )