Skip to content

Commit a3ca7c7

Browse files
remove failing test (temp)
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 9000666 commit a3ca7c7

File tree

1 file changed

+0
-39
lines changed

1 file changed

+0
-39
lines changed

tests/unit/test_sea_result_set.py

Lines changed: 0 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -88,45 +88,6 @@ def test_init_with_execute_response(
8888
assert isinstance(result_set.results, JsonQueue)
8989
assert result_set.results.data_array == []
9090

91-
def test_init_with_result_data(
92-
self,
93-
mock_connection,
94-
mock_sea_client,
95-
execute_response,
96-
mock_result_data,
97-
mock_manifest,
98-
):
99-
"""Test initializing SeaResultSet with result data."""
100-
with patch(
101-
"databricks.sql.result_set.SeaResultSetQueueFactory"
102-
) as mock_factory:
103-
mock_queue = Mock(spec=JsonQueue)
104-
mock_factory.build_queue.return_value = mock_queue
105-
106-
result_set = SeaResultSet(
107-
connection=mock_connection,
108-
execute_response=execute_response,
109-
sea_client=mock_sea_client,
110-
buffer_size_bytes=1000,
111-
arraysize=100,
112-
result_data=mock_result_data,
113-
manifest=mock_manifest,
114-
)
115-
116-
# Verify that the factory was called with the correct arguments
117-
mock_factory.build_queue.assert_called_once_with(
118-
mock_result_data,
119-
mock_manifest,
120-
str(execute_response.command_id.to_sea_statement_id()),
121-
description=execute_response.description,
122-
max_download_threads=mock_sea_client.max_download_threads,
123-
sea_client=mock_sea_client,
124-
lz4_compressed=execute_response.lz4_compressed,
125-
)
126-
127-
# Verify that the queue was set correctly
128-
assert result_set.results == mock_queue
129-
13091
def test_close(self, mock_connection, mock_sea_client, execute_response):
13192
"""Test closing a result set."""
13293
result_set = SeaResultSet(

0 commit comments

Comments
 (0)