correct sea res set tests

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

src/databricks/sql/result_set.py

@@ -19,7 +19,7 @@
 from databricks.sql.thrift_api.TCLIService import ttypes
 from databricks.sql.types import Row
 from databricks.sql.exc import Error, RequestError, CursorAlreadyClosedError
-from databricks.sql.utils import ColumnTable, ColumnQueue, JsonQueue
+from databricks.sql.utils import ColumnTable, ColumnQueue
 from databricks.sql.backend.types import CommandId, CommandState, ExecuteResponse
 
 logger = logging.getLogger(__name__)
@@ -484,7 +484,9 @@ def __init__(
 
     def _fill_results_buffer(self):
         """Fill the results buffer from the backend."""
-        raise NotImplementedError("fetchall_arrow is not implemented for SEA backend")
+        raise NotImplementedError(
+            "_fill_results_buffer is not implemented for SEA backend"
+        )
 
     def fetchone(self) -> Optional[Row]:
         """

tests/unit/test_sea_result_set.py

@@ -195,6 +195,7 @@ def test_fill_results_buffer_not_implemented(
         )
 
         with pytest.raises(
-            NotImplementedError, match="fetchone is not implemented for SEA backend"
+            NotImplementedError,
+            match="_fill_results_buffer is not implemented for SEA backend",
         ):
             result_set._fill_results_buffer()