@@ -210,6 +210,26 @@ index 0efe0877e9b..423d3b3d76d 100644
210
210
--
211
211
-- SELECT_HAVING
212
212
-- https://github.com/postgres/postgres/blob/REL_12_BETA2/src/test/regress/sql/select_having.sql
213
+ diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameAggregateSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameAggregateSuite.scala
214
+ index ea5e47ede55..cab59443c79 100644
215
+ --- a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameAggregateSuite.scala
216
+ +++ b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameAggregateSuite.scala
217
+ @@ -24,6 +24,7 @@ import scala.util.Random
218
+ import org.scalatest.matchers.must.Matchers.the
219
+
220
+ import org.apache.spark.SparkException
221
+ + import org.apache.spark.sql.comet.execution.shuffle.CometShuffleExchangeExec
222
+ import org.apache.spark.sql.execution.WholeStageCodegenExec
223
+ import org.apache.spark.sql.execution.adaptive.AdaptiveSparkPlanHelper
224
+ import org.apache.spark.sql.execution.aggregate.{HashAggregateExec, ObjectHashAggregateExec, SortAggregateExec}
225
+ @@ -756,6 +757,7 @@ class DataFrameAggregateSuite extends QueryTest
226
+
227
+ val exchangePlans = collect(aggPlan) {
228
+ case shuffle: ShuffleExchangeExec => shuffle
229
+ + case shuffle: CometShuffleExchangeExec => shuffle
230
+ }
231
+ assert(exchangePlans.length == 1)
232
+ }
213
233
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameJoinSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameJoinSuite.scala
214
234
index 56e9520fdab..917932336df 100644
215
235
--- a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameJoinSuite.scala
0 commit comments