@@ -417,7 +417,7 @@ index daef11ae4d6..9f3cc9181f2 100644
417417 assert(exchanges.size == 2)
418418 }
419419diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DynamicPartitionPruningSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DynamicPartitionPruningSuite.scala
420- index f33432ddb6f..99729e465e4 100644
420+ index f33432ddb6f..b7a5fd72f7d 100644
421421--- a/sql/core/src/test/scala/org/apache/spark/sql/DynamicPartitionPruningSuite.scala
422422+++ b/sql/core/src/test/scala/org/apache/spark/sql/DynamicPartitionPruningSuite.scala
423423@@ -22,6 +22,7 @@ import org.scalatest.GivenWhenThen
@@ -515,21 +515,13 @@ index f33432ddb6f..99729e465e4 100644
515515 }
516516 assert(subqueryBroadcastExecs.size === 1)
517517 subqueryBroadcastExecs.foreach { subqueryBroadcastExec =>
518- @@ -1698,7 +1718,8 @@ abstract class DynamicPartitionPruningV1Suite extends DynamicPartitionPruningDat
519- * Check the static scan metrics with and without DPP
520- */
521- test("static scan metrics",
522- - DisableAdaptiveExecution("DPP in AQE must reuse broadcast")) {
523- + DisableAdaptiveExecution("DPP in AQE must reuse broadcast"),
524- + IgnoreCometNativeDataFusion("https://github.com/apache/datafusion-comet/issues/3442")) {
525- withSQLConf(SQLConf.DYNAMIC_PARTITION_PRUNING_ENABLED.key -> "true",
526- SQLConf.DYNAMIC_PARTITION_PRUNING_REUSE_BROADCAST_ONLY.key -> "false",
527- SQLConf.EXCHANGE_REUSE_ENABLED.key -> "false") {
528- @@ -1729,6 +1750,8 @@ abstract class DynamicPartitionPruningV1Suite extends DynamicPartitionPruningDat
518+ @@ -1729,6 +1749,10 @@ abstract class DynamicPartitionPruningV1Suite extends DynamicPartitionPruningDat
529519 case s: BatchScanExec =>
530520 // we use f1 col for v2 tables due to schema pruning
531521 s.output.exists(_.exists(_.argString(maxFields = 100).contains("f1")))
532522+ case s: CometScanExec =>
523+ + s.output.exists(_.exists(_.argString(maxFields = 100).contains("fid")))
524+ + case s: CometNativeScanExec =>
533525+ s.output.exists(_.exists(_.argString(maxFields = 100).contains("fid")))
534526 case _ => false
535527 }
0 commit comments