Fix test

Author: wangyum

sql/core/src/test/scala/org/apache/spark/sql/SubquerySuite.scala

@@ -1879,27 +1879,26 @@ class SubquerySuite extends QueryTest with SharedSparkSession with AdaptiveSpark
   }
 
   test("SPARK-36280: Remove redundant aliases after RewritePredicateSubquery") {
-    sql("CREATE TABLE t1 USING parquet AS SELECT id AS a, id AS b, id AS c FROM range(10)")
-    sql("CREATE TABLE t2 USING parquet AS SELECT id AS x, id AS y FROM range(8)")
-    val df = sql(
-      """
-        |SELECT *
-        |FROM   t1
-        |WHERE  a IN (SELECT x
-        |             FROM   (SELECT x                         AS x,
-        |                            Rank()
-        |                              OVER (
-        |                                PARTITION BY x
-        |                                ORDER BY Sum(y) DESC) AS ranking
-        |                     FROM   t2
-        |                     GROUP  BY x) tmp1
-        |             WHERE  ranking <= 5)
-        |""".stripMargin)
-
-    df.collect()
-    val exchanges = collect(df.queryExecution.executedPlan) {
-      case s: ShuffleExchangeExec => s
+    withTable("t1", "t2") {
+      sql("CREATE TABLE t1 USING parquet AS SELECT id AS a, id AS b, id AS c FROM range(10)")
+      sql("CREATE TABLE t2 USING parquet AS SELECT id AS x, id AS y FROM range(8)")
+      val df = sql(
+        """
+          |SELECT *
+          |FROM   t1
+          |WHERE  a IN (SELECT x
+          |             FROM   (SELECT x AS x,
+          |                            RANK() OVER (PARTITION BY x ORDER BY SUM(y) DESC) AS ranking
+          |                     FROM   t2
+          |                     GROUP  BY x) tmp1
+          |             WHERE  ranking <= 5)
+          |""".stripMargin)
+
+      df.collect()
+      val exchanges = collect(df.queryExecution.executedPlan) {
+        case s: ShuffleExchangeExec => s
+      }
+      assert(exchanges.size === 1)
     }
-    assert(exchanges.size === 1)
   }
 }