Migrate jubjub benchmarks to criterion

jubjub/benches/fr_bench.rs

@@ -1,51 +1,58 @@
-#![feature(test)]
-
-extern crate test;
-
+use criterion::{criterion_group, criterion_main, Criterion};
 use jubjub::*;
-use test::Bencher;
 
-#[bench]
-fn bench_mul_assign(bencher: &mut Bencher) {
+fn bench_add_assign(c: &mut Criterion) {
     let mut n = Fr::one();
-    let b = -Fr::one();
-    bencher.iter(move || {
-        n *= &b;
+    let neg_one = -Fr::one();
+    c.bench_function("Fr add_assign", |b| {
+        b.iter(move || {
+            n += &neg_one;
+        })
     });
 }
 
-#[bench]
-fn bench_sub_assign(bencher: &mut Bencher) {
+fn bench_sub_assign(c: &mut Criterion) {
     let mut n = Fr::one();
-    let b = -Fr::one();
-    bencher.iter(move || {
-        n -= &b;
+    let neg_one = -Fr::one();
+    c.bench_function("Fr sub_assign", |b| {
+        b.iter(move || {
+            n -= &neg_one;
+        })
     });
 }
 
-#[bench]
-fn bench_add_assign(bencher: &mut Bencher) {
+fn bench_mul_assign(c: &mut Criterion) {
     let mut n = Fr::one();
-    let b = -Fr::one();
-    bencher.iter(move || {
-        n += &b;
+    let neg_one = -Fr::one();
+    c.bench_function("Fr mul_assign", |b| {
+        b.iter(move || {
+            n *= &neg_one;
+        })
     });
 }
 
-#[bench]
-fn bench_square_assign(bencher: &mut Bencher) {
+fn bench_square(c: &mut Criterion) {
     let n = Fr::one();
-    bencher.iter(move || n.square());
+    c.bench_function("Fr square", |b| b.iter(move || n.square()));
 }
 
-#[bench]
-fn bench_invert(bencher: &mut Bencher) {
+fn bench_invert(c: &mut Criterion) {
     let n = Fr::one();
-    bencher.iter(move || n.invert());
+    c.bench_function("Fr invert", |b| b.iter(move || n.invert()));
 }
 
-#[bench]
-fn bench_sqrt(bencher: &mut Bencher) {
+fn bench_sqrt(c: &mut Criterion) {
     let n = Fr::one().double().double();
-    bencher.iter(move || n.sqrt());
+    c.bench_function("Fr sqrt", |b| b.iter(move || n.sqrt()));
 }
+
+criterion_group!(
+    benches,
+    bench_add_assign,
+    bench_sub_assign,
+    bench_mul_assign,
+    bench_square,
+    bench_invert,
+    bench_sqrt,
+);
+criterion_main!(benches);