remove unimplemented method tests

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

Author: varun-edachali-dbx

src/databricks/sql/backend/thrift_backend.py

@@ -1232,8 +1232,6 @@ def fetch_results(
                 )
             )
 
-        from databricks.sql.utils import ThriftResultSetQueueFactory
-
         queue = ThriftResultSetQueueFactory.build_queue(
             row_set_type=resp.resultSetMetadata.resultFormat,
             t_row_set=resp.results,

src/databricks/sql/result_set.py

@@ -480,7 +480,6 @@ def __init__(
                 str(execute_response.command_id.to_sea_statement_id()),
                 description=execute_response.description,
                 max_download_threads=sea_client.max_download_threads,
-                ssl_options=sea_client.ssl_options,
                 sea_client=sea_client,
                 lz4_compressed=execute_response.lz4_compressed,
             )

src/databricks/sql/utils.py

@@ -119,7 +119,6 @@ def build_queue(
         description: Optional[List[Tuple[Any, ...]]] = None,
         schema_bytes: Optional[bytes] = None,
         max_download_threads: Optional[int] = None,
-        ssl_options: Optional[SSLOptions] = None,
         sea_client: Optional["SeaDatabricksClient"] = None,
         lz4_compressed: bool = False,
     ) -> ResultSetQueue:

tests/unit/test_sea_result_set.py

@@ -122,80 +122,3 @@ def test_close_when_connection_closed(
         mock_sea_client.close_command.assert_not_called()
         assert result_set.has_been_closed_server_side is True
         assert result_set.status == CommandState.CLOSED
-
-    def test_unimplemented_methods(
-        self, mock_connection, mock_sea_client, execute_response
-    ):
-        """Test that unimplemented methods raise NotImplementedError."""
-        result_set = SeaResultSet(
-            connection=mock_connection,
-            execute_response=execute_response,
-            sea_client=mock_sea_client,
-            buffer_size_bytes=1000,
-            arraysize=100,
-        )
-
-        # Test each unimplemented method individually with specific error messages
-        with pytest.raises(
-            NotImplementedError, match="fetchone is not implemented for SEA backend"
-        ):
-            result_set.fetchone()
-
-        with pytest.raises(
-            NotImplementedError, match="fetchmany is not implemented for SEA backend"
-        ):
-            result_set.fetchmany(10)
-
-        with pytest.raises(
-            NotImplementedError, match="fetchmany is not implemented for SEA backend"
-        ):
-            # Test with default parameter value
-            result_set.fetchmany()
-
-        with pytest.raises(
-            NotImplementedError, match="fetchall is not implemented for SEA backend"
-        ):
-            result_set.fetchall()
-
-        with pytest.raises(
-            NotImplementedError,
-            match="fetchmany_arrow is not implemented for SEA backend",
-        ):
-            result_set.fetchmany_arrow(10)
-
-        with pytest.raises(
-            NotImplementedError,
-            match="fetchall_arrow is not implemented for SEA backend",
-        ):
-            result_set.fetchall_arrow()
-
-        with pytest.raises(
-            NotImplementedError, match="fetchone is not implemented for SEA backend"
-        ):
-            # Test iteration protocol (calls fetchone internally)
-            next(iter(result_set))
-
-        with pytest.raises(
-            NotImplementedError, match="fetchone is not implemented for SEA backend"
-        ):
-            # Test using the result set in a for loop
-            for row in result_set:
-                pass
-
-    def test_fill_results_buffer_not_implemented(
-        self, mock_connection, mock_sea_client, execute_response
-    ):
-        """Test that _fill_results_buffer raises NotImplementedError."""
-        result_set = SeaResultSet(
-            connection=mock_connection,
-            execute_response=execute_response,
-            sea_client=mock_sea_client,
-            buffer_size_bytes=1000,
-            arraysize=100,
-        )
-
-        with pytest.raises(
-            NotImplementedError,
-            match="_fill_results_buffer is not implemented for SEA backend",
-        ):
-            result_set._fill_results_buffer()