bench(smt-subtree): add a benchmark for single-leaf subtrees

0xPolygonMiden · Oct 25, 2024 · fb0ff72 · fb0ff72
1 parent 9638969
commit fb0ff72
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/benches/smt-subtree.rs b/benches/smt-subtree.rs
@@ -9,12 +9,14 @@ use miden_crypto::{
 use rand_utils::prng_array;
 use winter_utils::Randomizable;
 
+const PAIR_COUNTS: [u64; 5] = [1, 64, 128, 192, 256];
+
 fn smt_subtree_even(c: &mut Criterion) {
     let mut seed = [0u8; 32];
 
     let mut group = c.benchmark_group("subtree8-even");
 
-    for pair_count in (64..=256).step_by(64) {
+    for pair_count in PAIR_COUNTS {
         let bench_id = BenchmarkId::from_parameter(pair_count);
         group.bench_with_input(bench_id, &pair_count, |b, &pair_count| {
             b.iter_batched(
@@ -64,7 +66,7 @@ fn smt_subtree_random(c: &mut Criterion) {
 
     let mut group = c.benchmark_group("subtree8-rand");
 
-    for pair_count in (64..=256).step_by(64) {
+    for pair_count in PAIR_COUNTS {
         let bench_id = BenchmarkId::from_parameter(pair_count);
         group.bench_with_input(bench_id, &pair_count, |b, &pair_count| {
             b.iter_batched(
@@ -94,10 +96,6 @@ fn smt_subtree_random(c: &mut Criterion) {
                         })
                         .collect();
                     leaves.sort();
-                    let before = leaves.len();
-                    leaves.dedup();
-                    let after = leaves.len();
-                    assert_eq!(before, after);
                     leaves
                 },
                 |leaves| {