From b8e241b1f2f05d0d9eaa872851e5b3078f55cfb7 Mon Sep 17 00:00:00 2001 From: Andy Grove Date: Fri, 30 Oct 2026 16:44:31 +0000 Subject: [PATCH] test: compare the V2 file insert's closed files as a set (#7335) (cherry picked from commit 36b432c896a4bf5e5bf0a9d26edad0375d978efa) --- .../org/apache/comet/parquet/CometParquetWriterSuite.scala | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/spark/src/test/scala/org/apache/comet/parquet/CometParquetWriterSuite.scala b/spark/src/test/scala/org/apache/comet/parquet/CometParquetWriterSuite.scala index e3a22569a24..af57ff0efac 100644 --- a/spark/src/test/scala/org/apache/comet/parquet/CometParquetWriterSuite.scala +++ b/spark/src/test/scala/org/apache/comet/parquet/CometParquetWriterSuite.scala @@ -2344,7 +2344,9 @@ class CometParquetWriterSuite extends CometParquetWriterTestBase { captureWritePlan(sql("INSERT INTO v2_file_target SELECT * FROM v2_file_source")) assertHasCometNativeWriteExec(plan) assert(RecordingFileSystem.created.size == 2, RecordingFileSystem.created) - assert(RecordingFileSystem.closed.asScala == RecordingFileSystem.created.asScala) + // The two tasks run concurrently, so their files close in any order. + assert( + RecordingFileSystem.closed.asScala.toSet == RecordingFileSystem.created.asScala.toSet) checkAnswer(spark.table("v2_file_target"), df) } }