From 7ff1e74cf9bf773c3546fe0a1a6918a3b732178e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tim=20Swe=C3=B1a=20=28Swast=29?= Date: Tue, 5 Aug 2025 16:50:48 -0500 Subject: [PATCH] chore: fix unit tests for `_read_gbq_colab` when Polars isn't installed --- tests/unit/pandas/io/test_api.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/unit/pandas/io/test_api.py b/tests/unit/pandas/io/test_api.py index 24ef51ad47..1e69fa9df3 100644 --- a/tests/unit/pandas/io/test_api.py +++ b/tests/unit/pandas/io/test_api.py @@ -14,10 +14,15 @@ from unittest import mock +import pytest + import bigframes.dataframe import bigframes.pandas.io.api as bf_io_api import bigframes.session +# _read_gbq_colab requires the polars engine. +pytest.importorskip("polars") + @mock.patch( "bigframes.pandas.io.api._set_default_session_location_if_possible_deferred_query"