use criterion::{criterion_group, criterion_main, Criterion};

use arrow2::array::Utf8Array;

fn add_benchmark(c: &mut Criterion) {
    (10..=20).step_by(2).for_each(|log2_size| {
        let size = 2usize.pow(log2_size);

        let array = Utf8Array::<i32>::from_trusted_len_values_iter(
            std::iter::repeat("aaa")
                .take(size / 2)
                .chain(std::iter::repeat("bbb").take(size / 2)),
        );

        c.bench_function(&format!("iter_values 2^{log2_size}"), |b| {
            b.iter(|| {
                for x in array.values_iter() {
                    assert_eq!(x.len(), 3);
                }
            })
        });

        c.bench_function(&format!("iter 2^{log2_size}"), |b| {
            b.iter(|| {
                for x in array.iter() {
                    assert_eq!(x.unwrap().len(), 3);
                }
            })
        });
    })
}

criterion_group!(benches, add_benchmark);
criterion_main!(benches);