From bd049b28c4f8f0e43f5d154dd4904da2b45f1769 Mon Sep 17 00:00:00 2001
From: Armin Becher <Armin Becher>
Date: Sun, 30 Apr 2023 22:51:45 +0200
Subject: [PATCH] test: add benchmark for remove coll

---
 benches/benches.rs | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/benches/benches.rs b/benches/benches.rs
index e725ae0..3e55238 100644
--- a/benches/benches.rs
+++ b/benches/benches.rs
@@ -157,6 +157,24 @@ fn criterion_benchmark(c: &mut Criterion) {
             criterion::BatchSize::SmallInput,
         )
     });
+    
+    // Remove
+    c.bench_function("grid_remove_row", |b| {
+        let grid = init_grid();
+        b.iter_batched(
+            || grid.clone(),
+            |mut g| g.remove_row(2),
+            criterion::BatchSize::SmallInput,
+        )
+    });
+    c.bench_function("grid_remove_col", |b| {
+        let grid = init_grid();
+        b.iter_batched(
+            || grid.clone(),
+            |mut g| g.remove_col(2),
+            criterion::BatchSize::SmallInput,
+        )
+    });
 }
 
 criterion_group!(benches, criterion_benchmark);