aboutsummaryrefslogtreecommitdiff
path: root/src/tests/aoc-2022-09/benches/main.rs
blob: bd5f92c5027b60536d557352b286dc6a6fe25a40 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
use aoc_2022_09::{gen_moves, part1, part2};
use criterion::{criterion_group, criterion_main, BatchSize, BenchmarkId, Criterion};
use rand::{rngs::StdRng, SeedableRng};

fn run_benches(c: &mut Criterion) {
    let mut rng = StdRng::seed_from_u64(42);
    for size in [100, 1000, 2000, 10_000, 50_000].iter() {
        c.bench_with_input(BenchmarkId::new("part1", size), size, |b, &n| {
            b.iter_batched_ref(
                || gen_moves(&mut rng, n).collect::<Vec<_>>(),
                |moves| part1(moves.iter()),
                BatchSize::SmallInput,
            )
        });

        c.bench_with_input(BenchmarkId::new("part2", size), size, |b, &n| {
            b.iter_batched_ref(
                || gen_moves(&mut rng, n).collect::<Vec<_>>(),
                |moves| part2(moves.iter()),
                BatchSize::SmallInput,
            )
        });
    }
}

criterion_group!(
    name = benches;
    config = Criterion::default().sample_size(10);
    targets = run_benches
);
criterion_main!(benches);