[SPARK-46896][PS][TESTS] Clean up the imports in pyspark.pandas.tests.{frame, series, groupby}.*

python/pyspark/pandas/tests/connect/frame/test_parity_attrs.py

@@ -16,16 +16,17 @@
 #
 import unittest
 
-from pyspark import pandas as ps
 from pyspark.pandas.tests.frame.test_attrs import FrameAttrsMixin
 from pyspark.testing.connectutils import ReusedConnectTestCase
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class FrameParityAttrsTests(FrameAttrsMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
-    @property
-    def psdf(self):
-        return ps.from_pandas(self.pdf)
+class FrameParityAttrsTests(
+    FrameAttrsMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
+    pass
 
 
 if __name__ == "__main__":

python/pyspark/pandas/tests/connect/frame/test_parity_axis.py

@@ -21,7 +21,11 @@
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class FrameParityAxisTests(FrameAxisMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
+class FrameParityAxisTests(
+    FrameAxisMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
     pass
 
 

python/pyspark/pandas/tests/connect/frame/test_parity_constructor.py

@@ -22,7 +22,9 @@
 
 
 class FrameParityConstructorTests(
-    FrameConstructorMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    FrameConstructorMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
     pass
 

python/pyspark/pandas/tests/connect/frame/test_parity_conversion.py

@@ -16,18 +16,17 @@
 #
 import unittest
 
-from pyspark import pandas as ps
 from pyspark.pandas.tests.frame.test_conversion import FrameConversionMixin
 from pyspark.testing.connectutils import ReusedConnectTestCase
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
 class FrameParityConversionTests(
-    FrameConversionMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    FrameConversionMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
-    @property
-    def psdf(self):
-        return ps.from_pandas(self.pdf)
+    pass
 
 
 if __name__ == "__main__":

python/pyspark/pandas/tests/connect/frame/test_parity_reindexing.py

@@ -16,18 +16,17 @@
 #
 import unittest
 
-from pyspark import pandas as ps
 from pyspark.pandas.tests.frame.test_reindexing import FrameReindexingMixin
 from pyspark.testing.connectutils import ReusedConnectTestCase
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
 class FrameParityReindexingTests(
-    FrameReindexingMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    FrameReindexingMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
-    @property
-    def psdf(self):
-        return ps.from_pandas(self.pdf)
+    pass
 
 
 if __name__ == "__main__":

python/pyspark/pandas/tests/connect/frame/test_parity_reshaping.py

@@ -21,7 +21,11 @@
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class FrameParityReshapingTests(FrameReshapingMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
+class FrameParityReshapingTests(
+    FrameReshapingMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
     pass
 
 

python/pyspark/pandas/tests/connect/frame/test_parity_spark.py

@@ -16,16 +16,17 @@
 #
 import unittest
 
-from pyspark import pandas as ps
 from pyspark.pandas.tests.frame.test_spark import FrameSparkMixin
 from pyspark.testing.connectutils import ReusedConnectTestCase
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class FrameParitySparkTests(FrameSparkMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
-    @property
-    def psdf(self):
-        return ps.from_pandas(self.pdf)
+class FrameParitySparkTests(
+    FrameSparkMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
+    pass
 
 
 if __name__ == "__main__":

python/pyspark/pandas/tests/connect/frame/test_parity_time_series.py

@@ -16,18 +16,17 @@
 #
 import unittest
 
-from pyspark import pandas as ps
 from pyspark.pandas.tests.frame.test_time_series import FrameTimeSeriesMixin
 from pyspark.testing.connectutils import ReusedConnectTestCase
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
 class FrameParityTimeSeriesTests(
-    FrameTimeSeriesMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    FrameTimeSeriesMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
-    @property
-    def psdf(self):
-        return ps.from_pandas(self.pdf)
+    pass
 
 
 if __name__ == "__main__":

python/pyspark/pandas/tests/connect/frame/test_parity_truncate.py

@@ -16,16 +16,17 @@
 #
 import unittest
 
-from pyspark import pandas as ps
 from pyspark.pandas.tests.frame.test_truncate import FrameTruncateMixin
 from pyspark.testing.connectutils import ReusedConnectTestCase
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class FrameParityTruncateTests(FrameTruncateMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
-    @property
-    def psdf(self):
-        return ps.from_pandas(self.pdf)
+class FrameParityTruncateTests(
+    FrameTruncateMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
+    pass
 
 
 if __name__ == "__main__":

python/pyspark/pandas/tests/connect/groupby/test_parity_aggregate.py

@@ -22,7 +22,9 @@
 
 
 class GroupbyParityAggregateTests(
-    GroupbyAggregateMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    GroupbyAggregateMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
     pass
 

python/pyspark/pandas/tests/connect/groupby/test_parity_apply_func.py

@@ -22,7 +22,9 @@
 
 
 class GroupbyParityApplyFuncTests(
-    GroupbyApplyFuncMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    GroupbyApplyFuncMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
     @unittest.skip("Test depends on SparkContext which is not supported from Spark Connect.")
     def test_apply_with_side_effect(self):

python/pyspark/pandas/tests/connect/groupby/test_parity_cumulative.py

@@ -22,7 +22,9 @@
 
 
 class GroupbyParityCumulativeTests(
-    GroupbyCumulativeMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    GroupbyCumulativeMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
     pass
 

python/pyspark/pandas/tests/connect/groupby/test_parity_describe.py

@@ -22,7 +22,9 @@
 
 
 class GroupbyParityDescribeTests(
-    GroupbyDescribeMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    GroupbyDescribeMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
     pass
 

python/pyspark/pandas/tests/connect/groupby/test_parity_groupby.py

@@ -22,7 +22,10 @@
 
 
 class GroupByParityTests(
-    GroupByTestsMixin, PandasOnSparkTestUtils, TestUtils, ReusedConnectTestCase
+    GroupByTestsMixin,
+    PandasOnSparkTestUtils,
+    TestUtils,
+    ReusedConnectTestCase,
 ):
     pass
 

python/pyspark/pandas/tests/connect/groupby/test_parity_head_tail.py

@@ -22,7 +22,9 @@
 
 
 class GroupbyParityHeadTailTests(
-    GroupbyHeadTailMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    GroupbyHeadTailMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
     pass
 

python/pyspark/pandas/tests/connect/groupby/test_parity_index.py

@@ -21,7 +21,11 @@
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class GroupbyParityIndexTests(GroupbyIndexMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
+class GroupbyParityIndexTests(
+    GroupbyIndexMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
     pass
 
 

python/pyspark/pandas/tests/connect/groupby/test_parity_missing_data.py

@@ -22,7 +22,9 @@
 
 
 class GroupbyParityMissingDataTests(
-    GroupbyMissingDataMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    GroupbyMissingDataMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
     pass
 

python/pyspark/pandas/tests/connect/series/test_parity_all_any.py

@@ -21,7 +21,11 @@
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class SeriesParityAllAnyTests(SeriesAllAnyMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
+class SeriesParityAllAnyTests(
+    SeriesAllAnyMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
     pass
 
 

python/pyspark/pandas/tests/connect/series/test_parity_arg_ops.py

@@ -21,7 +21,11 @@
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class SeriesParityArgOpsTests(SeriesArgOpsMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
+class SeriesParityArgOpsTests(
+    SeriesArgOpsMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
     pass
 
 

python/pyspark/pandas/tests/connect/series/test_parity_as_of.py

@@ -21,7 +21,11 @@
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class SeriesParityArgOpsTests(SeriesAsOfMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
+class SeriesParityArgOpsTests(
+    SeriesAsOfMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
     pass
 
 

python/pyspark/pandas/tests/connect/series/test_parity_as_type.py

@@ -21,7 +21,11 @@
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class SeriesParityAsTypeTests(SeriesAsTypeMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
+class SeriesParityAsTypeTests(
+    SeriesAsTypeMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
     pass
 
 

python/pyspark/pandas/tests/connect/series/test_parity_compute.py

@@ -21,7 +21,11 @@
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class SeriesParityComputeTests(SeriesComputeMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
+class SeriesParityComputeTests(
+    SeriesComputeMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
     pass
 
 

python/pyspark/pandas/tests/connect/series/test_parity_conversion.py

@@ -22,7 +22,9 @@
 
 
 class SeriesParityConversionTests(
-    SeriesConversionMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    SeriesConversionMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
     pass
 

python/pyspark/pandas/tests/connect/series/test_parity_cumulative.py

@@ -22,7 +22,9 @@
 
 
 class SeriesParityCumulativeTests(
-    SeriesCumulativeMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    SeriesCumulativeMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
     pass
 

python/pyspark/pandas/tests/connect/series/test_parity_index.py

@@ -21,7 +21,11 @@
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class SeriesParityIndexTests(SeriesIndexMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
+class SeriesParityIndexTests(
+    SeriesIndexMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
     pass
 
 

python/pyspark/pandas/tests/connect/series/test_parity_missing_data.py

@@ -22,7 +22,9 @@
 
 
 class SeriesParityMissingDataTests(
-    SeriesMissingDataMixin, PandasOnSparkTestUtils, ReusedConnectTestCase
+    SeriesMissingDataMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
 ):
     pass
 

python/pyspark/pandas/tests/connect/series/test_parity_series.py

@@ -21,7 +21,11 @@
 from pyspark.testing.pandasutils import PandasOnSparkTestUtils
 
 
-class SeriesParityTests(SeriesTestsMixin, PandasOnSparkTestUtils, ReusedConnectTestCase):
+class SeriesParityTests(
+    SeriesTestsMixin,
+    PandasOnSparkTestUtils,
+    ReusedConnectTestCase,
+):
     pass