Skip to content

Commit 5ce757c

Browse files
committed
Add metrics checking to CometExecSuite "CometBroadcastExchangeExec".
1 parent af4876b commit 5ce757c

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

spark/src/test/scala/org/apache/comet/exec/CometExecSuite.scala

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -474,6 +474,10 @@ class CometExecSuite extends CometTestBase {
474474
val expected = (0 until numParts).flatMap(_ => (0 until 5).map(i => i + 1)).sorted
475475

476476
assert(rowContents === expected)
477+
478+
val metrics = nativeBroadcast.metrics
479+
assert(metrics("numCoalescedBatches").value == 5L)
480+
assert(metrics("numCoalescedRows").value == 5L)
477481
}
478482
}
479483
}

0 commit comments

Comments
 (0)