correct patch paths

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

Author: varun-edachali-dbx

src/databricks/sql/backend/types.py

@@ -80,7 +80,6 @@ def from_thrift_state(
             return cls.CANCELLED
         else:
             return None
-    
 
     @classmethod
     def from_sea_state(cls, state: str) -> Optional["CommandState"]:
@@ -412,6 +411,7 @@ def to_hex_guid(self) -> str:
         else:
             return str(self.guid)
 
+
 @dataclass
 class ExecuteResponse:
     """Response from executing a SQL command."""

tests/unit/test_thrift_backend.py

@@ -1231,7 +1231,7 @@ def test_get_catalogs_calls_client_and_handle_execute_response(
             response, cursor_mock
         )
 
-    @patch("databricks.sql.backend.thrift_backend.ThriftResultSet")
+    @patch("databricks.sql.result_set.ThriftResultSet")
     @patch("databricks.sql.backend.thrift_backend.TCLIService.Client", autospec=True)
     def test_get_schemas_calls_client_and_handle_execute_response(
         self, tcli_service_class, mock_result_set
@@ -1273,7 +1273,7 @@ def test_get_schemas_calls_client_and_handle_execute_response(
             response, cursor_mock
         )
 
-    @patch("databricks.sql.backend.thrift_backend.ThriftResultSet")
+    @patch("databricks.sql.result_set.ThriftResultSet")
     @patch("databricks.sql.backend.thrift_backend.TCLIService.Client", autospec=True)
     def test_get_tables_calls_client_and_handle_execute_response(
         self, tcli_service_class, mock_result_set
@@ -1319,7 +1319,7 @@ def test_get_tables_calls_client_and_handle_execute_response(
             response, cursor_mock
         )
 
-    @patch("databricks.sql.backend.thrift_backend.ThriftResultSet")
+    @patch("databricks.sql.result_set.ThriftResultSet")
     @patch("databricks.sql.backend.thrift_backend.TCLIService.Client", autospec=True)
     def test_get_columns_calls_client_and_handle_execute_response(
         self, tcli_service_class, mock_result_set