Skip to content
This repository was archived by the owner on Mar 6, 2026. It is now read-only.

Commit 031adf1

Browse files
committed
fix(bigquery): update tests to expect timeout parameter
1 parent f57810c commit 031adf1

3 files changed

Lines changed: 4 additions & 0 deletions

File tree

tests/unit/job/test_query_pandas.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1023,5 +1023,6 @@ def test_query_job_to_geodataframe_delegation(wait_for_query):
10231023
int_dtype=DefaultPandasDTypes.INT_DTYPE,
10241024
float_dtype=None,
10251025
string_dtype=None,
1026+
timeout=None,
10261027
)
10271028
assert df is row_iterator.to_geodataframe.return_value

tests/unit/test_table.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5665,6 +5665,7 @@ def test_rowiterator_to_geodataframe_delegation(self, to_dataframe):
56655665
int_dtype=DefaultPandasDTypes.INT_DTYPE,
56665666
float_dtype=None,
56675667
string_dtype=None,
5668+
timeout=None,
56685669
)
56695670

56705671
self.assertIsInstance(df, geopandas.GeoDataFrame)

tests/unit/test_table_pandas.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -301,6 +301,7 @@ def test_rowiterator_to_geodataframe_with_default_dtypes(
301301
int_dtype=bigquery.enums.DefaultPandasDTypes.INT_DTYPE,
302302
float_dtype=None,
303303
string_dtype=None,
304+
timeout=None,
304305
)
305306
mock_geopandas.GeoDataFrame.assert_called_once_with(
306307
mock_df, crs="EPSG:4326", geometry="geo_col"
@@ -358,6 +359,7 @@ def test_rowiterator_to_geodataframe_with_custom_dtypes(
358359
int_dtype=custom_int_dtype,
359360
float_dtype=custom_float_dtype,
360361
string_dtype=custom_string_dtype,
362+
timeout=None,
361363
)
362364
mock_geopandas.GeoDataFrame.assert_called_once_with(
363365
mock_df, crs="EPSG:4326", geometry="geo_col"

0 commit comments

Comments
 (0)