Remove unused table in test.

apache · Mar 13, 2024 · 8b7946c · 8b7946c
1 parent 1367121
commit 8b7946c
Showing 1 changed file with 7 additions and 9 deletions.
diff --git a/spark/src/test/scala/org/apache/comet/exec/CometColumnarShuffleSuite.scala b/spark/src/test/scala/org/apache/comet/exec/CometColumnarShuffleSuite.scala
@@ -944,15 +944,13 @@ class CometShuffleSuite extends CometColumnarShuffleSuite {
       CometConf.COMET_EXEC_SHUFFLE_ENABLED.key -> "true",
       CometConf.COMET_COLUMNAR_SHUFFLE_ENABLED.key -> "true") {
       withParquetTable((0 until 10).map(i => (i, i % 5)), "tbl_a") {
-        withParquetTable((0 until 10).map(i => (i % 10, i + 2)), "tbl_b") {
-          val df = sql("SELECT * FROM tbl_a")
-          val shuffled = df
-            .select($"_1" + 1 as ("a"))
-            .filter($"a" > 4)
-            .repartition(10)
-            .sortWithinPartitions($"a")
-          checkSparkAnswerAndOperator(shuffled, classOf[ShuffleQueryStageExec])
-        }
+        val df = sql("SELECT * FROM tbl_a")
+        val shuffled = df
+          .select($"_1" + 1 as ("a"))
+          .filter($"a" > 4)
+          .repartition(10)
+          .sortWithinPartitions($"a")
+        checkSparkAnswerAndOperator(shuffled, classOf[ShuffleQueryStageExec])
       }
     }
   }