1 files changed, 17 insertions, 10 deletions

diff --git a/tests/test_pydora/test_utils.py b/tests/test_pydora/test_utils.py
index fc33a40..9d3bc78 100644
--- a/tests/test_pydora/test_utils.py
+++ b/tests/test_pydora/test_utils.py
@@ -10,23 +10,26 @@ from pydora.utils import iterate_forever
 
 
 class TestIterateForever(TestCase):
-
     def setUp(self):
         self.transport = Mock(side_effect=[InvalidAuthToken(), None])
         self.client = APIClient(self.transport, None, None, None, None)
         self.client._authenticate = Mock()
 
     def test_handle_missing_params_exception_due_to_missing_ad_tokens(self):
-        with patch.object(APIClient, 'get_playlist') as get_playlist_mock:
+        with patch.object(APIClient, "get_playlist") as get_playlist_mock:
             admock = patch.object(
-                APIClient, 'register_ad',
-                side_effect=ParameterMissing("ParameterMissing"))
+                APIClient,
+                "register_ad",
+                side_effect=ParameterMissing("ParameterMissing"),
+            )
 
             with admock:
                 station = Station.from_json(
-                    self.client, {'stationToken': 'token_mock'})
+                    self.client, {"stationToken": "token_mock"}
+                )
                 ad_mock = AdItem.from_json(
-                    self.client, {'station_id': 'id_mock'})
+                    self.client, {"station_id": "id_mock"}
+                )
                 get_playlist_mock.return_value = iter([ad_mock])
 
                 station_iter = iterate_forever(station.get_playlist)
@@ -36,15 +39,19 @@ class TestIterateForever(TestCase):
 
     def test_reraise_missing_params_exception(self):
         plmock = patch.object(
-            APIClient, 'get_playlist',
-            side_effect=ParameterMissing("ParameterMissing"))
+            APIClient,
+            "get_playlist",
+            side_effect=ParameterMissing("ParameterMissing"),
+        )
 
         with plmock as get_playlist_mock:
             with self.assertRaises(ParameterMissing):
                 station = Station.from_json(
-                    self.client, {'stationToken': 'token_mock'})
+                    self.client, {"stationToken": "token_mock"}
+                )
                 track_mock = PlaylistItem.from_json(
-                    self.client, {'token': 'token_mock'})
+                    self.client, {"token": "token_mock"}
+                )
                 get_playlist_mock.return_value = iter([track_mock])
 
                 station_iter = iterate_forever(station.get_playlist)