diff --git a/python/pyspark/pandas/tests/test_arrow_interface.py b/python/pyspark/pandas/tests/test_arrow_interface.py index d38462dbe9e1d..304c38c756c41 100644 --- a/python/pyspark/pandas/tests/test_arrow_interface.py +++ b/python/pyspark/pandas/tests/test_arrow_interface.py @@ -21,7 +21,7 @@ pyarrow_requirement_message, ) from pyspark import pandas as ps -from pyspark.testing.sqlutils import SQLTestUtils +from pyspark.testing.pandasutils import PandasOnSparkTestCase import pandas as pd @@ -60,7 +60,7 @@ def test_spark_arrow_c_streamer_arrow_consumer(self): self.assertEqual(result, expected) -class ArrowInterfaceTests(ArrowInterfaceTestsMixin, SQLTestUtils): +class ArrowInterfaceTests(ArrowInterfaceTestsMixin, PandasOnSparkTestCase): pass