From 3761d2ed47518a6f5d9141190d634792ca0b9f4c Mon Sep 17 00:00:00 2001 From: Kumar Ujjawal Date: Tue, 12 May 2026 18:11:59 +0530 Subject: [PATCH] fix(bench): avoid OOM in `array_replace` bench --- .../benches/array_expression.rs | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/datafusion/functions-nested/benches/array_expression.rs b/datafusion/functions-nested/benches/array_expression.rs index ad9f565f4d643..71bb939238aab 100644 --- a/datafusion/functions-nested/benches/array_expression.rs +++ b/datafusion/functions-nested/benches/array_expression.rs @@ -23,27 +23,22 @@ use std::hint::black_box; fn criterion_benchmark(c: &mut Criterion) { // Construct large arrays for benchmarking - let array_len = 100000000; + let array_len = 100_000; let array = (0..array_len).map(|_| lit(2_i64)).collect::>(); let list_array = make_array(vec![make_array(array); 3]); let from_array = make_array(vec![lit(2_i64); 3]); let to_array = make_array(vec![lit(-2_i64); 3]); - let expected_array = list_array.clone(); - // Benchmark array functions c.bench_function("array_replace", |b| { b.iter(|| { - assert_eq!( - array_replace_all( - list_array.clone(), - from_array.clone(), - to_array.clone() - ), - *black_box(&expected_array) - ) + black_box(array_replace_all( + list_array.clone(), + from_array.clone(), + to_array.clone(), + )) }) }); }