skip connect tests if deps are not there

HyukjinKwon · Sep 25, 2024 · 0cdc148 · 0cdc148
1 parent e2d2ab5
commit 0cdc148
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/pyspark/sql/tests/test_connect_compatibility.py b/python/pyspark/sql/tests/test_connect_compatibility.py
@@ -18,6 +18,7 @@
 import unittest
 import inspect
 
+from pyspark.testing.connectutils import should_test_connect, connect_requirement_message
 from pyspark.testing.sqlutils import ReusedSQLTestCase
 from pyspark.sql.classic.dataframe import DataFrame as ClassicDataFrame
 from pyspark.sql.connect.dataframe import DataFrame as ConnectDataFrame
@@ -172,6 +173,7 @@ def check_missing_methods(classic_cls, connect_cls, cls_name, expected_missing_m
         )
 
 
+@unittest.skipIf(not should_test_connect, connect_requirement_message)
 class ConnectCompatibilityTests(ConnectCompatibilityTestsMixin, ReusedSQLTestCase):
     pass