diff --git a/src/stats.rs b/src/stats.rs index 422706c..e92d21a 100644 --- a/src/stats.rs +++ b/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); } }