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

Commit 57b826c

Browse files
fix more tests
1 parent 06f9dbd commit 57b826c

File tree

5 files changed

+269
-239
lines changed

5 files changed

+269
-239
lines changed

tests/system/small/bigquery/test_struct.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,5 +58,5 @@ def test_struct_from_dataframe(columns_arg):
5858
bbq.struct(srs.struct.explode()).to_pandas(),
5959
check_index_type=False,
6060
check_dtype=False,
61-
check_namesbool=False, # None vs nan version dependent
61+
check_names=False, # None vs nan version dependent
6262
)

tests/system/small/operations/test_timedeltas.py

Lines changed: 19 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -508,10 +508,11 @@ def test_timedelta_filtering(session):
508508
pd.Timestamp("2025-01-01 01:00:00"),
509509
pd.Timestamp("2025-01-01 02:00:00"),
510510
pd.Timestamp("2025-01-01 03:00:00"),
511-
]
511+
],
512+
dtype=dtypes.TIMESTAMP_DTYPE,
512513
)
513514
bf_series = session.read_pandas(pd_series)
514-
timestamp = pd.Timestamp("2025-01-01, 00:00:01")
515+
timestamp = pd.Timestamp("2025-01-01, 00:00:01", tz="UTC")
515516

516517
actual_result = bf_series[
517518
((bf_series - timestamp) > pd.Timedelta(1, "h"))
@@ -526,16 +527,22 @@ def test_timedelta_filtering(session):
526527
def test_timedelta_ordering(session):
527528
pd_df = pd.DataFrame(
528529
{
529-
"col_1": [
530-
pd.Timestamp("2025-01-01 01:00:00"),
531-
pd.Timestamp("2025-01-01 02:00:00"),
532-
pd.Timestamp("2025-01-01 03:00:00"),
533-
],
534-
"col_2": [
535-
pd.Timestamp("2025-01-01 01:00:02"),
536-
pd.Timestamp("2025-01-01 02:00:01"),
537-
pd.Timestamp("2025-01-01 02:59:59"),
538-
],
530+
"col_1": pd.Series(
531+
[
532+
pd.Timestamp("2025-01-01 01:00:00"),
533+
pd.Timestamp("2025-01-01 02:00:00"),
534+
pd.Timestamp("2025-01-01 03:00:00"),
535+
],
536+
dtype=dtypes.TIMESTAMP_DTYPE,
537+
),
538+
"col_2": pd.Series(
539+
[
540+
pd.Timestamp("2025-01-01 01:00:02"),
541+
pd.Timestamp("2025-01-01 02:00:01"),
542+
pd.Timestamp("2025-01-01 02:59:59"),
543+
],
544+
dtype=dtypes.TIMESTAMP_DTYPE,
545+
),
539546
}
540547
)
541548
bf_df = session.read_pandas(pd_df)

0 commit comments

Comments
 (0)