From deef6fd91186a460f877db79339bc463d27ef581 Mon Sep 17 00:00:00 2001
From: Zach Dziura <zcdziura@gmail.com>
Date: Mon, 5 Oct 2015 08:17:53 -0400
Subject: [PATCH] Added benchmarking

---
 src/lib.rs   |  1 +
 src/prime.rs | 18 ++++++++++++++++++
 2 files changed, 19 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index fd67033..751da29 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,5 +1,6 @@
 #![feature(augmented_assignments)]
 #![feature(core)]
+#![feature(test)]
 
 extern crate core;
 extern crate ramp;
diff --git a/src/prime.rs b/src/prime.rs
index 867b9fe..80224b0 100644
--- a/src/prime.rs
+++ b/src/prime.rs
@@ -211,3 +211,21 @@ fn rewrite(candidate: &Int) -> (Int, Int) {
 
     (s, d)
 }
+
+#[cfg(test)]
+mod tests {
+    extern crate test;
+
+    use super::*;
+    use self::test::Bencher;
+
+    #[bench]
+    fn bench_prime_gen_1024(b: &mut Bencher) {
+        b.iter(|| Prime::new(1024));
+    }
+
+    #[bench]
+    fn bench_prime_gen_2048(b: &mut Bencher) {
+        b.iter(|| Prime::new(2048));
+    }
+}