formatting (black)

as in CONTRIBUTING.md

Signed-off-by: varun-edachali-dbx <[email protected]>

Author: varun-edachali-dbx

src/databricks/sql/client.py

@@ -235,10 +235,13 @@ def read(self) -> Optional[OAuthToken]:
             catalog,
             schema,
             _use_arrow_native_complex_types,
-            **kwargs
+            **kwargs,
+        )
+
+        logger.info(
+            "Successfully opened connection with session "
+            + str(self.get_session_id_hex())
         )
-        
-        logger.info("Successfully opened connection with session " + str(self.get_session_id_hex()))
 
         self.use_inline_params = self._set_use_inline_params_with_warning(
             kwargs.get("use_inline_params", False)

src/databricks/sql/session.py

@@ -26,13 +26,13 @@ def __init__(
     ) -> None:
         """
         Create a session to a Databricks SQL endpoint or a Databricks cluster.
-        
+
         This class handles all session-related behavior and communication with the backend.
         """
         self.open = False
         self.host = server_hostname
         self.port = kwargs.get("_port", 443)
-        
+
         auth_provider = get_python_sql_connector_auth_provider(
             server_hostname, **kwargs
         )
@@ -143,4 +143,4 @@ def close(self) -> None:
         except Exception as e:
             logger.error(f"Attempt to close session raised a local exception: {e}")
 
-        self.open = False 
+        self.open = False

tests/unit/test_client.py

@@ -146,7 +146,7 @@ def test_closing_result_set_hard_closes_commands(self):
         mock_session = Mock()
         mock_session.open = True
         type(mock_connection).session = PropertyMock(return_value=mock_session)
-        
+
         result_set = client.ResultSet(
             mock_connection, mock_results_response, mock_thrift_backend
         )

tests/unit/test_session.py

@@ -11,7 +11,7 @@
 
 class SessionTestSuite(unittest.TestCase):
     """
-    Unit tests for Session functionality 
+    Unit tests for Session functionality
     """
 
     PACKAGE_NAME = "databricks.sql"
@@ -184,4 +184,4 @@ def test_finalizer_closes_abandoned_connection(self, mock_client_class):
 
 
 if __name__ == "__main__":
-    unittest.main() 
+    unittest.main()

tests/unit/test_thrift_backend.py

@@ -86,7 +86,9 @@ def test_make_request_checks_thrift_status_code(self):
 
     def _make_type_desc(self, type):
         return ttypes.TTypeDesc(
-            types=[ttypes.TTypeEntry(primitiveEntry=ttypes.TPrimitiveTypeEntry(type=type))]
+            types=[
+                ttypes.TTypeEntry(primitiveEntry=ttypes.TPrimitiveTypeEntry(type=type))
+            ]
         )
 
     def _make_fake_thrift_backend(self):