use const_array_init::make_const_arr;

make_const_arr!(ARR1, [i32; 3], div_2);

const fn div_2(n: usize) -> i32 { n as i32 / 2 }

#[test]
fn function_test() {
    assert_eq!(ARR1, [0, 0, 1]);
}

make_const_arr!(ARR2, [i32; 3], |i| i as i32 / 2);

#[test]
fn closure_test() {
    assert_eq!(ARR2, [0, 0, 1]);
}

make_const_arr!(ARR3, [i32; 3], |_| 4 / 2);

#[test]
fn easy_closure_test() {
    assert_eq!(ARR3, [2, 2, 2]);
}

make_const_arr!(ARR4, [i32; 3], |i| { 
    let n = 2; 
    i as i32 / n
});

#[test]
fn advanced_closure_test() {
    assert_eq!(ARR4, [0, 0, 1]);
}

#[derive(Debug, PartialEq, Eq)]
struct User {
    id: u32,
}

make_const_arr!(ARR5, [User; 3], |i| User { id: i as u32 });

#[test]
fn super_advanced_closure_test() {
    assert_eq!(ARR5, [User { id: 0 }, User { id: 1 }, User { id: 2 },]);
}

make_const_arr!(ARR6, [User; 3], |_| User { id: 1 as u32 });

#[test]
fn super_advanced_closure_test_empty_closure() {
    assert_eq!(ARR6, [User { id: 1 }, User { id: 1 }, User { id: 1 },]);
}


const fn create_user(n: usize) -> User {
    User { id: n as u32 }
}

make_const_arr!(ARR7, [User; 3], create_user);

#[test]
fn super_advanced_func_test() {
    assert_eq!(ARR7, [User { id: 0 }, User { id: 1 }, User { id: 2 },]);
}