1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

/// Create new 2d array with ones on the diagonal and zeros elsewhere
///
/// # Examples
///
/// ```
/// use arr_rs::prelude::*;
///
/// let arr: Array<i32> = array_arange!(0, 4).unwrap();
/// assert_eq!(array!([0, 1, 2, 3, 4]).unwrap(), arr);
/// let arr: Array<i32> = array_arange!(0, 7, 2).unwrap();
/// assert_eq!(array!([0, 2, 4, 6]).unwrap(), arr);
/// ```
#[macro_export]
macro_rules! array_arange {
    ($n:expr, $m:expr) => {
        array_arange!($n, $m, 1)
    };
    ($n:expr, $m:expr, $k:expr) => {{
        Array::arange($n, $m, Some($k))
    }};
}