From b59f8018c6458af4060a4f20e6bdcc76a9da1c20 Mon Sep 17 00:00:00 2001 From: Liang-Chi Hsieh Date: Thu, 14 Nov 2024 12:45:30 -0800 Subject: [PATCH] test: Restore one test in CometExecSuite by adding COMET_SHUFFLE_MODE config --- .../apache/comet/exec/CometExecSuite.scala | 22 ++++++++++--------- 1 file changed, 12 insertions(+), 10 deletions(-) diff --git a/spark/src/test/scala/org/apache/comet/exec/CometExecSuite.scala b/spark/src/test/scala/org/apache/comet/exec/CometExecSuite.scala index 78f59cbea..0cef78f75 100644 --- a/spark/src/test/scala/org/apache/comet/exec/CometExecSuite.scala +++ b/spark/src/test/scala/org/apache/comet/exec/CometExecSuite.scala @@ -100,17 +100,19 @@ class CometExecSuite extends CometTestBase { } test("ShuffleQueryStageExec could be direct child node of CometBroadcastExchangeExec") { - val table = "src" - withTable(table) { - withView("lv_noalias") { - sql(s"CREATE TABLE $table (key INT, value STRING) USING PARQUET") - sql(s"INSERT INTO $table VALUES(238, 'val_238')") + withSQLConf(CometConf.COMET_SHUFFLE_MODE.key -> "jvm") { + val table = "src" + withTable(table) { + withView("lv_noalias") { + sql(s"CREATE TABLE $table (key INT, value STRING) USING PARQUET") + sql(s"INSERT INTO $table VALUES(238, 'val_238')") - sql( - "CREATE VIEW lv_noalias AS SELECT myTab.* FROM src " + - "LATERAL VIEW explode(map('key1', 100, 'key2', 200)) myTab LIMIT 2") - val df = sql("SELECT * FROM lv_noalias a JOIN lv_noalias b ON a.key=b.key"); - checkSparkAnswer(df) + sql( + "CREATE VIEW lv_noalias AS SELECT myTab.* FROM src " + + "LATERAL VIEW explode(map('key1', 100, 'key2', 200)) myTab LIMIT 2") + val df = sql("SELECT * FROM lv_noalias a JOIN lv_noalias b ON a.key=b.key"); + checkSparkAnswer(df) + } } } }