Skip to content

Commit 166b093

Browse files
committed
Return scalar instead of array from generator
1 parent 1e7301c commit 166b093

File tree

1 file changed

+10
-16
lines changed

1 file changed

+10
-16
lines changed

arrow/benches/zip_kernels.rs

Lines changed: 10 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -241,14 +241,14 @@ fn bench_zip_on_string_view_scalars(
241241
.as_str(),
242242
);
243243

244-
let null_scalar = input_generator_1.generate_null();
244+
let null_scalar = input_generator_1.generate_null_scalar();
245245

246-
let non_null_scalar_1 = input_generator_1.generate();
247-
let non_null_scalar_2 = input_generator_2.generate();
246+
let non_null_scalar_1 = input_generator_1.generate_scalar();
247+
let non_null_scalar_2 = input_generator_2.generate_scalar();
248248

249249
let masks = mask_cases(ARRAY_LEN);
250250

251-
for (description, truthy, falsy) in &[
251+
for (description, truthy, falsy) in [
252252
("null_vs_non_null_scalar", &null_scalar, &non_null_scalar_1),
253253
(
254254
"non_null_scalar_vs_null_scalar",
@@ -257,13 +257,7 @@ fn bench_zip_on_string_view_scalars(
257257
),
258258
("non_nulls_scalars", &non_null_scalar_1, &non_null_scalar_2),
259259
] {
260-
bench_zip_input_on_all_masks(
261-
description,
262-
&mut group,
263-
&masks,
264-
&Scalar::new(truthy),
265-
&Scalar::new(falsy),
266-
);
260+
bench_zip_input_on_all_masks(description, &mut group, &masks, truthy, falsy);
267261
}
268262
group.finish();
269263
}
@@ -278,16 +272,16 @@ impl GenerateStringView {
278272
fn name(&self) -> &str {
279273
self.description.as_str()
280274
}
281-
fn generate_null(&self) -> ArrayRef {
282-
new_null_array(&DataType::Utf8View, 1)
275+
fn generate_null_scalar(&self) -> Scalar<ArrayRef> {
276+
Scalar::new(new_null_array(&DataType::Utf8View, 1))
283277
}
284278

285-
fn generate(&self) -> ArrayRef {
286-
Arc::new(create_string_view_array_with_fixed_len(
279+
fn generate_scalar(&self) -> Scalar<ArrayRef> {
280+
Scalar::new(Arc::new(create_string_view_array_with_fixed_len(
287281
1,
288282
0.0,
289283
self.str_len,
290-
))
284+
)))
291285
}
292286
}
293287

0 commit comments

Comments
 (0)