Skip to content

Commit da17b7d

Browse files
committed
update plans
1 parent 3da662d commit da17b7d

1 file changed

Lines changed: 9 additions & 14 deletions

File tree

spark/src/test/scala/org/apache/comet/exec/CometAggregateSuite.scala

Lines changed: 9 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -680,7 +680,8 @@ class CometAggregateSuite extends CometTestBase with AdaptiveSparkPlanHelper {
680680
test("partialMerge - cnt distinct + sum") {
681681
withTempDir(dir => {
682682
withSQLConf("spark.comet.enabled" -> "false") {
683-
sql("""
683+
withView("t") {
684+
sql("""
684685
CREATE OR REPLACE TEMP VIEW t (v, v1, i) AS
685686
VALUES
686687
('c', 'a', 1),
@@ -699,28 +700,22 @@ class CometAggregateSuite extends CometTestBase with AdaptiveSparkPlanHelper {
699700
('c3', 'a3', 2),
700701
('c4', 'a4', 2)
701702
""")
702-
sql("select * from t")
703-
.repartition(3)
704-
.write
705-
.mode("overwrite")
706-
.parquet(dir.getAbsolutePath)
703+
sql("select * from t")
704+
.repartition(3)
705+
.write
706+
.mode("overwrite")
707+
.parquet(dir.getAbsolutePath)
708+
}
707709
}
708710

709711
withSQLConf(
710712
CometConf.COMET_EXEC_SHUFFLE_ENABLED.key -> "true",
711713
"spark.comet.exec.shuffle.fallbackToColumnar" -> "false",
712714
"spark.comet.cast.allowIncompatible" -> "true",
713715
"spark.sql.adaptive.enabled" -> "false",
714-
"spark.comet.explain.native.enabled" -> "true",
715716
"spark.comet.enabled" -> "true",
716717
"spark.comet.expression.Cast.allowIncompatible" -> "true",
717-
"spark.comet.exec.shuffle.enableFastEncoding" -> "true",
718-
"spark.comet.exec.shuffle.enabled" -> "true",
719-
"spark.comet.explainFallback.enabled" -> "true",
720-
CometConf.COMET_NATIVE_SCAN_IMPL.key -> "native_iceberg_compat",
721-
"spark.shuffle.manager" ->
722-
"org.apache.spark.sql.comet.execution.shuffle.CometShuffleManager",
723-
"spark.comet.logFallbackReasons.enabled" -> "true") {
718+
CometConf.COMET_NATIVE_SCAN_IMPL.key -> "native_iceberg_compat") {
724719
spark.read.parquet(dir.getAbsolutePath).createOrReplaceTempView("t2")
725720
checkSparkAnswerAndOperator("SELECT i, sum(v1), count(distinct v) FROM t2 group by i")
726721
}

0 commit comments

Comments
 (0)