PEP8 Test Cleanup

1 files changed, 12 insertions, 10 deletions

diff --git a/tests/test_pandora/test_client.py b/tests/test_pandora/test_client.py
index cf486e0..f7f0574 100644
--- a/tests/test_pandora/test_client.py
+++ b/tests/test_pandora/test_client.py
@@ -1,7 +1,7 @@
 from unittest import TestCase
 
+from pandora import errors
 from pandora.client import APIClient, BaseAPIClient
-from pandora.errors import InvalidAuthToken, ParameterMissing
 from pandora.py2compat import Mock, call, patch
 from tests.test_pandora.test_models import TestAdItem
 
@@ -46,7 +46,7 @@ class TestAPIClientLogin(TestCase):
 class TestCallingAPIClient(TestCase):
 
     def test_call_should_retry_on_token_error(self):
-        transport = Mock(side_effect=[InvalidAuthToken(), None])
+        transport = Mock(side_effect=[errors.InvalidAuthToken(), None])
 
         client = BaseAPIClient(transport, None, None, None)
         client._authenticate = Mock()
@@ -59,7 +59,7 @@ class TestCallingAPIClient(TestCase):
 
     def test_ad_support_enabled_parameters(self):
         with patch.object(APIClient, '__call__') as playlist_mock:
-            transport = Mock(side_effect=[InvalidAuthToken(), None])
+            transport = Mock(side_effect=[errors.InvalidAuthToken(), None])
 
             client = APIClient(transport, None, None, None, None)
             client._authenticate = Mock()
@@ -97,8 +97,9 @@ class TestGettingQualities(TestCase):
 class TestGettingAds(TestCase):
 
     def test_get_ad_item_(self):
-        with patch.object(APIClient, '__call__', return_value=TestAdItem.JSON_DATA) as ad_metadata_mock:
-            transport = Mock(side_effect=[InvalidAuthToken(), None])
+        with patch.object(APIClient, '__call__',
+                return_value=TestAdItem.JSON_DATA) as ad_metadata_mock:
+            transport = Mock(side_effect=[errors.InvalidAuthToken(), None])
 
             client = APIClient(transport, None, None, None, None)
             client._authenticate = Mock()
@@ -108,14 +109,15 @@ class TestGettingAds(TestCase):
             assert ad_item.ad_token == 'token_mock'
 
             ad_metadata_mock.assert_has_calls([call("ad.getAdMetadata",
-                                                      adToken='token_mock',
-                                                      returnAdTrackingTokens=True,
-                                                      supportAudioAds=True)])
+                                                    adToken='token_mock',
+                                                    returnAdTrackingTokens=True,
+                                                    supportAudioAds=True)])
 
     def test_get_ad_item_with_no_station_id_specified_raises_exception(self):
-        transport = Mock(side_effect=[InvalidAuthToken(), None])
+        transport = Mock(side_effect=[errors.InvalidAuthToken(), None])
 
         client = APIClient(transport, None, None, None, None)
         client.get_ad_metadata = Mock()
 
-        self.assertRaises(ParameterMissing, client.get_ad_item, '', 'token_mock')
+        self.assertRaises(
+                errors.ParameterMissing, client.get_ad_item, '', 'token_mock')