diff --git a/tests/system/test_read_gbq.py b/tests/system/test_read_gbq.py index 0187b69d..f9358ef6 100644 --- a/tests/system/test_read_gbq.py +++ b/tests/system/test_read_gbq.py @@ -547,7 +547,13 @@ def test_default_dtypes( ): if use_bqstorage_api not in use_bqstorage_apis: pytest.skip(f"use_bqstorage_api={use_bqstorage_api} not supported.") - result = read_gbq(query, use_bqstorage_api=use_bqstorage_api) + # the parameter useQueryCache=False is used in the following function call + # to avoid a failing test due to cached data that may be out of order. + result = read_gbq( + query, + use_bqstorage_api=use_bqstorage_api, + configuration={"query": {"useQueryCache": False}}, + ) pandas.testing.assert_frame_equal(result, expected) diff --git a/tests/system/test_read_gbq_with_bqstorage.py b/tests/system/test_read_gbq_with_bqstorage.py index 84435b9f..cfb31ea8 100644 --- a/tests/system/test_read_gbq_with_bqstorage.py +++ b/tests/system/test_read_gbq_with_bqstorage.py @@ -50,7 +50,7 @@ def test_large_results(random_dataset, method_under_test): total_amount, passenger_count, trip_distance - FROM `bigquery-public-data.new_york_taxi_trips.tlc_green_trips_2014` + FROM `bigquery-public-data.new_york.tlc_green_trips_2014` -- Select non-null rows for no-copy conversion from Arrow to pandas. WHERE total_amount IS NOT NULL AND passenger_count IS NOT NULL