Skip to content

Commit 2de3a37

Browse files
committed
Remove pandas.testing
1 parent b249bac commit 2de3a37

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

python/pyspark/sql/tests.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)