Reformat with black

Author: Greg Leclercq

integration_tests/fixtures.py

@@ -9,26 +9,23 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
+from __future__ import absolute_import, division, print_function
 
-from contextlib import closing
 import os
-import requests
 import socket
 import subprocess
 import time
-from uuid import uuid4
 import xml.etree.ElementTree as ET
+from contextlib import closing
+from uuid import uuid4
 
 import click
+import prestodb.logging
 import pytest
-
+import requests
 from prestodb.client import PrestoQuery, PrestoRequest
 from prestodb.constants import DEFAULT_PORT
 from prestodb.exceptions import TimeoutError
-import prestodb.logging
 
 
 logger = prestodb.logging.get_logger(__name__)

integration_tests/test_dbapi.py

@@ -9,15 +9,12 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
+from __future__ import absolute_import, division, print_function
 
 import fixtures
-from fixtures import run_presto
 import prestodb
 import pytest
-
+from fixtures import run_presto
 from prestodb.transaction import IsolationLevel
 
 

prestodb/client.py

@@ -109,9 +109,9 @@ def __init__(self, id, stats, warnings, info_uri, next_uri, rows, columns=None):
 
     def __repr__(self):
         return (
-            'PrestoStatus('
-            'id={}, stats={{...}}, warnings={}, info_uri={}, next_uri={}, rows=<count={}>'
-            ')'.format(
+            "PrestoStatus("
+            "id={}, stats={{...}}, warnings={}, info_uri={}, next_uri={}, rows=<count={}>"
+            ")".format(
                 self.id,
                 len(self.warnings),
                 self.info_uri,
@@ -413,10 +413,10 @@ def process(self, http_response):
         self._next_uri = response.get("nextUri")
 
         return PrestoStatus(
-            id=response['id'],
-            stats=response['stats'],
-            warnings=response.get('warnings', []),
-            info_uri=response['infoUri'],
+            id=response["id"],
+            stats=response["stats"],
+            warnings=response.get("warnings", []),
+            info_uri=response["infoUri"],
             next_uri=self._next_uri,
             rows=response.get("data", []),
             columns=response.get("columns"),
@@ -511,7 +511,7 @@ def execute(self):
         self.query_id = status.id
         self._stats.update({u"queryId": self.query_id})
         self._stats.update(status.stats)
-        self._warnings = getattr(status, 'warnings', [])
+        self._warnings = getattr(status, "warnings", [])
         if status.next_uri is None:
             self._finished = True
         self._result = PrestoResult(self, status.rows)