From 08b53db76651b389bac3632d5230bc48c1ef92d0 Mon Sep 17 00:00:00 2001
From: Vinzent Steinberg <Vinzent.Steinberg@gmail.com>
Date: Thu, 4 May 2017 11:17:50 +0200
Subject: [PATCH] Add statistical test

---
 Cargo.toml | 13 ++++++++-----
 src/lib.rs | 24 +++++++++++++++++++++++-
 2 files changed, 31 insertions(+), 6 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index 2c03390..e562795 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,12 +1,15 @@
 [package]
 authors = ["Vinzent Steinberg <Vinzent.Steinberg@gmail.com>"]
-name = "average"
-version = "0.1.0"
-license = "MIT/Apache-2.0"
-repository = "https://github.com/vks/average"
+categories = ["science", "no-std"]
 description = "Calculate the average of a sequence and its error iteratively"
 keywords = ["statistics", "stats"]
-categories = ["science", "no-std"]
+license = "MIT/Apache-2.0"
+name = "average"
+repository = "https://github.com/vks/average"
+version = "0.1.0"
 
 [dependencies]
 conv = "0.3"
+
+[dev-dependencies]
+rand = "0.3"
diff --git a/src/lib.rs b/src/lib.rs
index 760d52b..9314db9 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,6 +1,8 @@
 #![no_std]
 
 extern crate conv;
+#[cfg(test)] extern crate rand;
+#[cfg(test)] #[macro_use] extern crate std;
 
 use conv::ApproxFrom;
 
@@ -76,6 +78,14 @@ impl core::iter::FromIterator<f64> for Average {
     }
 }
 
+    macro_rules! assert_almost_eq {
+        ($a:expr, $b:expr, $prec:expr) => (
+            if ($a - $b).abs() > $prec {
+                panic!(format!("assertion failed: `abs(left - right) < {:e}`, (left: `{}`, right: `{}`)", $prec, $a, $b));
+            }
+        );
+    }
+
 #[cfg(test)]
 mod tests {
     use super::*;
@@ -99,6 +109,18 @@ mod tests {
         assert_eq!(a.avg(), 3.0);
         assert_eq!(a.len(), 5);
         assert_eq!(a.var(), 2.5);
-        assert!((a.err() - f64::sqrt(0.5)).abs() < 1e-16);
+        assert_almost_eq!(a.err(), f64::sqrt(0.5), 1e-16);
+    }
+
+    #[test]
+    fn average_normal_distribution() {
+        use rand::distributions::{Normal, IndependentSample};
+        let normal = Normal::new(2.0, 3.0);
+        let mut a = Average::new();
+        for _ in 0..1000000 {
+            a.add(normal.ind_sample(&mut ::rand::thread_rng()));
+        }
+        assert_almost_eq!(a.avg(), 2.0, 1e-2);
+        assert_almost_eq!(a.var().sqrt(), 3.0, 1e-2);
     }
 }