Simple Array Sum (https://www.hackerrank.com/challenges/simple-array-sum/problem)

main.rs

use std::io;

fn main() {
    let arr = get_arr(get_n());
    println!("{}", sum_arr(&arr));
}

fn get_n() -> usize {
    let mut s = String::new();
    io::stdin().read_line(&mut s).unwrap_or(0);
    s.trim().parse::<usize>().unwrap_or(0)
}

fn get_arr(n :usize) -> Vec<u16> {
    let mut s = String::new();
    io::stdin().read_line(&mut s).unwrap_or(0);

    let mut arr = vec![0; n];
    for (i, value) in s.trim().split_whitespace().enumerate() {
        if i > n {
            break;
        }
        arr[i] = value.parse::<u16>().unwrap_or(0);
    }
    arr
}

fn sum_arr(arr :&[u16]) -> u16 {
    let mut sum :u16 = 0;
    for value in arr.iter() {
        sum += value;
    }
    sum
}

#[test]
fn test_sum_arr() {
    assert_eq!(10, sum_arr(&[1, 2, 3, 4]));
    assert_eq!(5, sum_arr(&[1, 2, 2]));
    assert_eq!(3, sum_arr(&[1, 1, 1]));
}