[SPARK-50311][TESTS][FOLLOWUP] Fix test name properly

### What changes were proposed in this pull request? This is minor followup for apache#48843 to fix test name properly from Python file ### Why are the changes needed? Use proper test name ### Does this PR introduce _any_ user-facing change? No, it's test only ### How was this patch tested? CI ### Was this patch authored or co-authored using generative AI tooling? No Closes apache#49199 from itholic/SPARK-50311-followup-2. Authored-by: Haejoon Lee <haejoon.lee@databricks.com> Signed-off-by: Haejoon Lee <haejoon.lee@databricks.com>
witgo · Jan 3, 2025 · 032e531 · 032e531
1 parent 9087d84
commit 032e531
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/pyspark/sql/tests/connect/test_parity_job_cancellation.py b/python/pyspark/sql/tests/connect/test_parity_job_cancellation.py
@@ -57,7 +57,7 @@ def test_interrupt_all(self):
 
 if __name__ == "__main__":
     import unittest
-    from pyspark.sql.tests.connect.test_parity_serde import *  # noqa: F401
+    from pyspark.sql.tests.connect.test_parity_job_cancellation import *  # noqa: F401
 
     try:
         import xmlrunner  # type: ignore[import]

diff --git a/python/pyspark/sql/tests/test_job_cancellation.py b/python/pyspark/sql/tests/test_job_cancellation.py
@@ -172,7 +172,7 @@ class JobCancellationTests(JobCancellationTestsMixin, ReusedSQLTestCase):
 
 
 if __name__ == "__main__":
-    from pyspark.sql.tests.test_session import *  # noqa: F401
+    from pyspark.sql.tests.test_job_cancellation import *  # noqa: F401
 
     try:
         import xmlrunner