diff --git a/backends-clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseColumnarShuffleAQESuite.scala b/backends-clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseColumnarShuffleAQESuite.scala index 0a858a35d9fa4..c1297d42309d1 100644 --- a/backends-clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseColumnarShuffleAQESuite.scala +++ b/backends-clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseColumnarShuffleAQESuite.scala @@ -161,8 +161,7 @@ class GlutenClickHouseColumnarShuffleAQESuite val adaptiveSparkPlanExec = collectWithSubqueries(df.queryExecution.executedPlan) { case adaptive: AdaptiveSparkPlanExec => adaptive } - assert(adaptiveSparkPlanExec.size == 3) - assert(adaptiveSparkPlanExec(1) == adaptiveSparkPlanExec(2)) + assert(adaptiveSparkPlanExec.size == 2) } } } diff --git a/backends-clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHousePreferSpillColumnarShuffleAQESuite.scala b/backends-clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHousePreferSpillColumnarShuffleAQESuite.scala index b7655f57d51b4..c7a35dc811bff 100644 --- a/backends-clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHousePreferSpillColumnarShuffleAQESuite.scala +++ b/backends-clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHousePreferSpillColumnarShuffleAQESuite.scala @@ -162,8 +162,7 @@ class GlutenClickHousePreferSpillColumnarShuffleAQESuite val adaptiveSparkPlanExec = collectWithSubqueries(df.queryExecution.executedPlan) { case adaptive: AdaptiveSparkPlanExec => adaptive } - assert(adaptiveSparkPlanExec.size == 3) - assert(adaptiveSparkPlanExec(1) == adaptiveSparkPlanExec(2)) + assert(adaptiveSparkPlanExec.size == 2) } } } diff --git a/backends-clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseTPCHColumnarShuffleParquetAQESuite.scala b/backends-clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseTPCHColumnarShuffleParquetAQESuite.scala index 32efabc9472e3..f710333d60c55 100644 --- a/backends-clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseTPCHColumnarShuffleParquetAQESuite.scala +++ b/backends-clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseTPCHColumnarShuffleParquetAQESuite.scala @@ -294,8 +294,7 @@ class GlutenClickHouseTPCHColumnarShuffleParquetAQESuite val adaptiveSparkPlanExec = collectWithSubqueries(df.queryExecution.executedPlan) { case adaptive: AdaptiveSparkPlanExec => adaptive } - assert(adaptiveSparkPlanExec.size == 3) - assert(adaptiveSparkPlanExec(1) == adaptiveSparkPlanExec(2)) + assert(adaptiveSparkPlanExec.size == 2) } } diff --git a/gluten-celeborn/clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala b/gluten-celeborn/clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala index 4b4b62135f769..44200782ce2a6 100644 --- a/gluten-celeborn/clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala +++ b/gluten-celeborn/clickhouse/src/test/scala/io/glutenproject/execution/GlutenClickHouseRSSColumnarShuffleAQESuite.scala @@ -172,8 +172,7 @@ class GlutenClickHouseRSSColumnarShuffleAQESuite val adaptiveSparkPlanExec = collectWithSubqueries(df.queryExecution.executedPlan) { case adaptive: AdaptiveSparkPlanExec => adaptive } - assert(adaptiveSparkPlanExec.size == 3) - assert(adaptiveSparkPlanExec(1) == adaptiveSparkPlanExec(2)) + assert(adaptiveSparkPlanExec.size == 2) } }