From 5e11332ef5d0fdc1f7bd89cf49f1df8a4a345de6 Mon Sep 17 00:00:00 2001 From: 0lai0 Date: Fri, 3 Apr 2026 15:14:55 +0800 Subject: [PATCH] test: parameterize file count in Native_datafusion metrics test --- .../test/scala/org/apache/comet/exec/CometExecSuite.scala | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/spark/src/test/scala/org/apache/comet/exec/CometExecSuite.scala b/spark/src/test/scala/org/apache/comet/exec/CometExecSuite.scala index f7d8179391..187cf3e07f 100644 --- a/spark/src/test/scala/org/apache/comet/exec/CometExecSuite.scala +++ b/spark/src/test/scala/org/apache/comet/exec/CometExecSuite.scala @@ -2210,9 +2210,11 @@ class CometExecSuite extends CometTestBase { } test("Native_datafusion reports correct files and bytes scanned") { + val inputFiles = 2 + withTempDir { dir => val path = new java.io.File(dir, "test_metrics").getAbsolutePath - spark.range(100).repartition(2).write.mode("overwrite").parquet(path) + spark.range(100).repartition(inputFiles).write.mode("overwrite").parquet(path) withSQLConf( CometConf.COMET_ENABLED.key -> "true", @@ -2236,8 +2238,8 @@ class CometExecSuite extends CometTestBase { val numFiles = scanNode.metrics("numFiles").value assert( - numFiles == 2, - s"Expected exactly 2 files to be scanned, but got metrics reporting $numFiles") + numFiles == inputFiles, + s"Expected exactly $inputFiles files to be scanned, but got metrics reporting $numFiles") } } }