ndarray 0.4.7

extern crate num;
extern crate ndarray;

use ndarray::{arr1, arr2, Axis};
use ndarray::OwnedArray;
use num::{Num, Complex};

fn c<T: Clone + Num>(re: T, im: T) -> Complex<T> {
    Complex::new(re, im)
}

#[test]
fn complex_mat_mul()
{
    let a = arr2(&[[c(3., 4.), c(2., 0.)], [c(0., -2.), c(3., 0.)]]);
    let b = (&a * c(3., 0.)).map(|c| 5. * c / c.norm());
    println!("{:>8.2}", b);
    let e = OwnedArray::eye(2);
    let r = a.mat_mul(&e);
    println!("{}", a);
    assert_eq!(r, a);
    assert_eq!(a.mean(Axis(0)), arr1(&[c(1.5, 1.), c(2.5, 0.)]));
}