Update playlists.py

Author: BillaSpace

tests/async/playlists.py

@@ -3,33 +3,48 @@
 import time
 from youtubesearchpython.__future__ import *
 
-def print_json(data):
+def pretty_print(data, fn_name, elapsed):
     print(json.dumps(data, indent=2, ensure_ascii=False))
-    print("-" * 60)
+    print(f"\n⏱ {fn_name} took {elapsed:.3f} seconds\n{'-'*60}\n")
 
-def print_timing(fn_name, elapsed):
-    print(f"⏱ {fn_name}: {elapsed:.3f}s")
-    print("=" * 60)
-
-async def timed(fn_name, coro):
+async def timed_call(fn_name, coro):
     start = time.perf_counter()
     result = await coro
     elapsed = time.perf_counter() - start
-    return fn_name, elapsed, result
+    return result, fn_name, elapsed
 
 async def main():
     url1 = "https://www.youtube.com/playlist?list=PLRBp0Fe2GpgmsW46rJyudVFlY6IYjFBIK"
     url2 = "https://www.youtube.com/watch?v=bplUXwTTgbI&list=PL6edxAMqu2xfxgbf7Q09hSg1qCMfDI7IZ"
 
-    for fn, coro in [
-        ("Playlist.get", Playlist.get(url1)),
-        ("Playlist.getInfo", Playlist.getInfo(url1)),
-        ("Playlist.getVideos", Playlist.getVideos(url1)),
-        ("Playlist.get (cached)", Playlist.get(url1)),
-        ("Playlist.get (video+playlist URL)", Playlist.get(url2)),
-    ]:
-        name, elapsed, result = await timed(fn, coro)
-        print_timing(name, elapsed)
-        print_json(result)
+    playlist, fn, t = await timed_call(
+        "Playlist.get",
+        Playlist.get(url1),
+    )
+    pretty_print(playlist, fn, t)
+
+    playlistInfo, fn, t = await timed_call(
+        "Playlist.getInfo",
+        Playlist.getInfo(url1),
+    )
+    pretty_print(playlistInfo, fn, t)
+
+    playlistVideos, fn, t = await timed_call(
+        "Playlist.getVideos",
+        Playlist.getVideos(url1),
+    )
+    pretty_print(playlistVideos, fn, t)
+
+    playlist, fn, t = await timed_call(
+        "Playlist.get",
+        Playlist.get(url1),
+    )
+    pretty_print(playlist, fn, t)
+
+    playlist, fn, t = await timed_call(
+        "Playlist.get (video+playlist URL)",
+        Playlist.get(url2),
+    )
+    pretty_print(playlist, fn, t)
 
 asyncio.run(main())