Skip to content

Commit 86e972e

Browse files
committed
fix tests
1 parent 6569723 commit 86e972e

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

packages/google-cloud-bigquery/tests/unit/test_query_results_format_arrow.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,8 @@
1616
import unittest
1717
from unittest import mock
1818

19+
import pytest
20+
1921
from google.cloud.bigquery import _job_helpers
2022
from google.cloud.bigquery.client import Client
2123
from google.cloud.bigquery.table import RowIterator, _EmptyRowIterator
@@ -522,6 +524,7 @@ def test_job_helpers_query_and_wait_accepts_enums(self):
522524
)
523525

524526
def test_row_iterator_to_dataframe_iterable_when_format_is_arrow(self):
527+
pytest.importorskip("pandas")
525528
mock_client = mock.MagicMock()
526529
mock_client._ensure_bqstorage_client.return_value = mock.MagicMock()
527530

@@ -544,6 +547,7 @@ def test_row_iterator_to_dataframe_iterable_when_format_is_arrow(self):
544547
self.assertEqual(dfs, ["df_chunk"])
545548

546549
def test_row_iterator_to_dataframe_when_format_is_arrow(self):
550+
pytest.importorskip("pandas")
547551
mock_client = mock.MagicMock()
548552
mock_client._ensure_bqstorage_client.return_value = mock.MagicMock()
549553

0 commit comments

Comments
 (0)