File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed
Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -3977,7 +3977,6 @@ def test_vectorized_udf_timestamps_respect_session_timezone(self):
39773977
39783978 def test_nondeterministic_udf (self ):
39793979 # Test that nondeterministic UDFs are evaluated only once in chained UDF evaluations
3980- from pandas .testing import assert_series_equal
39813980 from pyspark .sql .functions import udf , pandas_udf , col
39823981
39833982 @pandas_udf ('double' )
@@ -3989,7 +3988,7 @@ def plus_ten(v):
39893988 result1 = df .withColumn ('plus_ten(rand)' , plus_ten (df ['rand' ])).toPandas ()
39903989
39913990 self .assertEqual (random_udf .deterministic , False )
3992- assert_series_equal (result1 ['plus_ten(rand)' ], result1 ['rand' ] + 10 , check_names = False )
3991+ self . assertTrue (result1 ['plus_ten(rand)' ]. equals ( result1 ['rand' ] + 10 ) )
39933992
39943993 def test_nondeterministic_udf_in_aggregate (self ):
39953994 from pyspark .sql .functions import pandas_udf , sum
You can’t perform that action at this time.
0 commit comments