Adding unit tests

src/stats.rs

@@ -74,5 +74,45 @@ impl<'a> NewStatsCalculator<'a> {
     //     mode
     // }
 }
+
+#[cfg(test)]
+mod test {
+    use super::*;
+
+    #[test]
+    fn test_get_length() {
+        let mut data = vec![1, 2, 3, 4, 5, 6];
+        let mut test = NewStatsCalculator::from_vec(&mut data);
+        test.get_length();
+        assert_eq!(test.length, 6);
+    }
+
+    #[test]
+    fn test_get_total() {
+        let mut data = vec![1, 2, 3, 4, 5, 6];
+        let mut test = NewStatsCalculator::from_vec(&mut data);
+        test.get_length();
+        assert_eq!(test.total, 21);
+    }
+
+    #[test]
+    fn test_get_mean() {
+        let mut data = vec![1, 2, 3, 4, 5, 6];
+        let test = NewStatsCalculator::from_vec(&mut data);
+        assert_eq!(test.get_mean(), 3.5);
+    }
+
+    #[test]
+    fn test_get_median_even() {
+        let mut data = vec![1, 2, 3, 4, 5, 6];
+        let mut test = NewStatsCalculator::from_vec(&mut data);
+        assert_eq!(test.get_median(), 3.5);
+    }
+
+    #[test]
+    fn test_get_median_odd() {
+        let mut data = vec![1, 2, 3, 4, 5, 6, 7];
+        let mut test = NewStatsCalculator::from_vec(&mut data);
+        assert_eq!(test.get_median(), 4.0);
     }
 }