flake

Author: maingoh

deepomatic/api/exceptions.py

@@ -26,35 +26,26 @@
 from tenacity import RetryError
 from requests import Response
 
-###############################################################################
 
 class DeepomaticException(Exception):
     def __init__(self, msg):
         super(DeepomaticException, self).__init__(msg)
 
-###############################################################################
-
 
 class CredentialsNotFound(DeepomaticException):
     pass
 
 
-###############################################################################
-
 class UnimplementedException(DeepomaticException):
     def __init__(self, msg):
         super(UnimplementedException, self).__init__(msg)
 
 
-###############################################################################
-
 class NoData(DeepomaticException):
     def __init__(self):
         super(NoData, self).__init__("No data !! You may need to call '.retrieve()' ?")
 
 
-###############################################################################
-
 class BadStatus(DeepomaticException):
     """
     Thrown when HTTP response status_code < 200 or status_code >= 300.
@@ -93,8 +84,6 @@ class ServerError(BadStatus):
     pass
 
 
-###############################################################################
-
 class TaskError(DeepomaticException):
     def __init__(self, task):
         self.task = task
@@ -106,8 +95,6 @@ def get_task_id(self):
         return self.task['id']
 
 
-###############################################################################
-
 class TaskTimeout(DeepomaticException):
     def __init__(self, task, retry_error=None):
         self.task = task
@@ -120,9 +107,6 @@ def get_task_id(self):
         return self.task['id']
 
 
-###############################################################################
-
-
 class HTTPRetryError(RetryError):
     def __str__(self):
         last_exception = self.last_attempt.exception(timeout=0)

tests/test_client.py

@@ -352,9 +352,12 @@ def test_retry_network_failure(self):
         with pytest.raises(HTTPRetryError) as exc:
             print(spec.data())  # does make a http call
 
-        assert str(exc.value).startswith("""
-        Last attempt was an exception <class 'requests.exceptions.ConnectionError'> "HTTPConnectionPool(host='invalid-domain.deepomatic.com', port=80): Max retries exceeded with url: /v0.7/recognition/public/imagenet-inception-v3/ (Caused by NewConnectionError('<urllib3.connection.HTTPConnection object at
-        """.strip())
+        assert str(exc.value).startswith(
+            """Last attempt was an exception <class 'requests.exceptions.ConnectionError'> \""""
+            """HTTPConnectionPool(host='invalid-domain.deepomatic.com', port=80): Max retries exceeded with url: """
+            """/v0.7/recognition/public/imagenet-inception-v3/ (Caused by NewConnectionError"""
+            """('<urllib3.connection.HTTPConnection object at"""
+        )
         assert str(exc.value).endswith("""
         >: Failed to establish a new connection: [Errno -2] Name or service not known',))"
         """.strip())