fix regression (#1259)

andygrove · web-flow · commit d52038e883e2 · 2025-01-10T11:31:48.000-07:00
diff --git a/spark/src/test/scala/org/apache/comet/exec/CometAggregateSuite.scala b/spark/src/test/scala/org/apache/comet/exec/CometAggregateSuite.scala
@@ -108,7 +108,7 @@ class CometAggregateSuite extends CometTestBase with AdaptiveSparkPlanHelper {
         val cometShuffles = collect(df2.queryExecution.executedPlan) {
           case _: CometShuffleExchangeExec => true
         }
-        if (shuffleMode == "jvm") {
+        if (shuffleMode == "jvm" || shuffleMode == "auto") {
           assert(cometShuffles.length == 1)
         } else {
           // we fall back to Spark for shuffle because we do not support

Original file line number	Diff line number	Diff line change
`@@ -108,7 +108,7 @@ class CometAggregateSuite extends CometTestBase with AdaptiveSparkPlanHelper {`
`108`	`108`	`val cometShuffles = collect(df2.queryExecution.executedPlan) {`
`109`	`109`	`case _: CometShuffleExchangeExec => true`
`110`	`110`	`}`
`111`		`- if (shuffleMode == "jvm") {`
	`111`	`+ if (shuffleMode == "jvm" \|\| shuffleMode == "auto") {`
`112`	`112`	`assert(cometShuffles.length == 1)`
`113`	`113`	`} else {`
`114`	`114`	`// we fall back to Spark for shuffle because we do not support`