feat: add types to args

tests/helpers.py

@@ -1,9 +1,12 @@
 import base64
+from typing import Union
 
 import requests
 
+from spotipy import Spotify
 
-def get_spotify_playlist(spotify_object, playlist_name, username):
+
+def get_spotify_playlist(spotify_object: Spotify, playlist_name: str, username: str):
     playlists = spotify_object.user_playlists(username)
     while playlists:
         for item in playlists['items']:
@@ -12,5 +15,5 @@ def get_spotify_playlist(spotify_object, playlist_name, username):
         playlists = spotify_object.next(playlists)
 
 
-def get_as_base64(url):
+def get_as_base64(url: Union[str, bytes]) -> str:
     return base64.b64encode(requests.get(url).content).decode("utf-8")

tests/unit/test_oauth.py

@@ -13,7 +13,7 @@ patch = mock.patch
 DEFAULT = mock.DEFAULT
 
 
-def _make_fake_token(expires_at, expires_in, scope):
+def _make_fake_token(expires_at: int, expires_in: int, scope: str):
     return dict(
         expires_at=expires_at,
         expires_in=expires_in,