formatting (black)

Author: varun-edachali-dbx

tests/unit/test_client.py

@@ -209,11 +209,11 @@ def test_context_manager_closes_cursor(self):
         mock_close.assert_called_once_with()
 
         cursor = client.Cursor(Mock(), Mock())
-        cursor.close = Mock() 
+        cursor.close = Mock()
 
         try:
             with self.assertRaises(KeyboardInterrupt):
-                with cursor: 
+                with cursor:
                     raise KeyboardInterrupt("Simulated interrupt")
         finally:
             cursor.close.assert_called()

tests/unit/test_session.py

@@ -146,11 +146,11 @@ def test_socket_timeout_passthrough(self, mock_client_class):
     @patch("%s.session.ThriftDatabricksClient" % PACKAGE_NAME)
     def test_configuration_passthrough(self, mock_client_class):
         mock_session_config = Mock()
-        
+
         # Create a mock SessionId that will be returned by open_session
         mock_session_id = SessionId(BackendType.THRIFT, b"\x22", b"\x33")
         mock_client_class.return_value.open_session.return_value = mock_session_id
-        
+
         databricks.sql.connect(
             session_configuration=mock_session_config, **self.DUMMY_CONNECTION_ARGS
         )
@@ -171,11 +171,11 @@ def test_initial_namespace_passthrough(self, mock_client_class):
         databricks.sql.connect(
             **self.DUMMY_CONNECTION_ARGS, catalog=mock_cat, schema=mock_schem
         )
-        
+
         # Check that open_session was called with the correct catalog and schema as positional arguments
         call_args = mock_client_class.return_value.open_session.call_args[0]
-        self.assertEqual(call_args[1], mock_cat)  # catalog is second positional argument
-        self.assertEqual(call_args[2], mock_schem)  # schema is third positional argument
+        self.assertEqual(call_args[1], mock_cat)
+        self.assertEqual(call_args[2], mock_schem)
 
     @patch("%s.session.ThriftDatabricksClient" % PACKAGE_NAME)
     def test_finalizer_closes_abandoned_connection(self, mock_client_class):