@@ -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):
526527def 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