Skip to content

Commit 746df87

Browse files
skip some pyarrow dependent tests if no pa
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent ac52428 commit 746df87

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/unit/test_sea_queue.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -204,6 +204,7 @@ def test_build_queue_json_array(self, json_manifest, sample_data):
204204
assert isinstance(queue, JsonQueue)
205205
assert queue.data_array == sample_data
206206

207+
@pytest.mark.skipif(pa is None, reason="pyarrow not installed")
207208
def test_build_queue_arrow_stream(
208209
self, arrow_manifest, ssl_options, mock_sea_client, description
209210
):
@@ -333,6 +334,7 @@ def test_convert_to_thrift_link_no_headers(self, sample_external_link_no_headers
333334

334335
@patch("databricks.sql.backend.sea.queue.ResultFileDownloadManager")
335336
@patch("databricks.sql.backend.sea.queue.logger")
337+
@pytest.mark.skipif(pa is None, reason="pyarrow not installed")
336338
def test_init_with_valid_initial_link(
337339
self,
338340
mock_logger,
@@ -487,6 +489,7 @@ def test_hybrid_disposition_with_attachment(
487489

488490
@patch("databricks.sql.backend.sea.queue.ResultFileDownloadManager")
489491
@patch.object(SeaCloudFetchQueue, "_create_next_table", return_value=None)
492+
@pytest.mark.skipif(pa is None, reason="pyarrow not installed")
490493
def test_hybrid_disposition_with_external_links(
491494
self,
492495
mock_create_table,

0 commit comments

Comments
 (0)