1825. Finding MK Average

[icealtria]

rust test case

#[cfg(test)]
mod tests {
    use crate::MKAverage;

    #[test]
    fn case1() {
        let mut obj = MKAverage::new(3, 1);
        obj.add_element(3);
        obj.add_element(1);
        assert_eq!(obj.calculate_mk_average(), -1);
        obj.add_element(10);       // current elements are [3,1,10]
        assert_eq!(obj.calculate_mk_average(), 3);
        obj.add_element(5);
        obj.add_element(5);
        obj.add_element(5);
        assert_eq!(obj.calculate_mk_average(), 5);
    }
    #[test]
    fn case2() {
        let mut obj = MKAverage::new(6,1);
        obj.add_element(3);
        obj.add_element(1);
        obj.add_element(12);
        obj.add_element(5);
        obj.add_element(3);
        obj.add_element(4);
        // obj.add_element(3);
        assert_eq!(obj.calculate_mk_average(), 3);
    }
}