@@ -382,12 +382,12 @@ def test_df_sr_binop(psr, colnames, binary_op):
382382 data = [[3.0 , 2.0 , 5.0 ], [3.0 , None , 5.0 ], [6.0 , 7.0 , np .nan ]]
383383 data = dict (zip (colnames , data , strict = True ))
384384
385- gsr = cudf .Series (psr ).astype ("float64 " )
385+ gsr = cudf .Series (psr ).astype ("Float64 " )
386386
387- gdf = cudf .DataFrame (data )
388- pdf = gdf .to_pandas (nullable = True )
387+ gdf = cudf .DataFrame (data ). astype ( "Float64" )
388+ pdf = gdf .to_pandas ()
389389
390- psr = gsr .to_pandas (nullable = True )
390+ psr = gsr .to_pandas ()
391391
392392 try :
393393 expect = binary_op (pdf , psr )
@@ -399,12 +399,12 @@ def test_df_sr_binop(psr, colnames, binary_op):
399399 with pytest .raises (ValueError ):
400400 binary_op (gsr , gdf )
401401 else :
402- got = binary_op (gdf , gsr ). to_pandas ( nullable = True )
403- assert_eq (expect , got , check_dtype = False , check_like = True )
402+ got = binary_op (gdf , gsr )
403+ assert_eq (expect , got )
404404
405405 expect = binary_op (psr , pdf )
406- got = binary_op (gsr , gdf ). to_pandas ( nullable = True )
407- assert_eq (expect , got , check_dtype = False , check_like = True )
406+ got = binary_op (gsr , gdf )
407+ assert_eq (expect , got )
408408
409409
410410@pytest .mark .parametrize (
0 commit comments