@@ -88,45 +88,6 @@ def test_init_with_execute_response(
88
88
assert isinstance (result_set .results , JsonQueue )
89
89
assert result_set .results .data_array == []
90
90
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
-
130
91
def test_close (self , mock_connection , mock_sea_client , execute_response ):
131
92
"""Test closing a result set."""
132
93
result_set = SeaResultSet (
0 commit comments