extern crate peroxide;
#[allow(unused_imports)]
use peroxide::fuga::*;
#[cfg(feature = "nc")]
#[test]
fn test_inverse() {
for i in 5..21 {
let df = DataFrame::read_nc_by_header(
&format!("test_data/rand_mat/randmat_{}.nc", i),
vec!["m", "inv"],
)
.unwrap();
let a_vec: Vec<f64> = df["m"].to_vec();
let b_vec: Vec<f64> = df["inv"].to_vec();
let a: Matrix = matrix(a_vec, i, i, Col);
let b: Matrix = matrix(b_vec, i, i, Col);
let c = a.inv();
assert_eq!(b, c);
}
}
#[cfg(feature = "nc")]
#[test]
fn test_matmul() {
for i in 5..21 {
let df = DataFrame::read_nc_by_header(
&format!("test_data/rand_mat/randmat_{}.nc", i),
vec!["m", "mm"],
)
.unwrap();
let a_vec: Vec<f64> = df["m"].to_vec();
let b_vec: Vec<f64> = df["mm"].to_vec();
let a: Matrix = matrix(a_vec, i, i, Col);
let b: Matrix = matrix(b_vec, i, i, Col);
let c = &a * &a;
assert_eq!(b, c);
}
}
#[cfg(feature = "nc")]
#[test]
fn test_matvecmul() {
for i in 5..21 {
let df = DataFrame::read_nc_by_header(
&format!("test_data/rand_mat/randmat_{}.nc", i),
vec!["m", "v", "mv"],
)
.unwrap();
let a_vec: Vec<f64> = df["m"].to_vec();
let a: Matrix = matrix(a_vec, i, i, Col);
let b: Vec<f64> = df["v"].to_vec();
let c: Vec<f64> = df["mv"].to_vec();
let d = &a * &b;
assert!(eq_vec(&c, &d, 1e-8));
}
}
#[cfg(feature = "nc")]
#[test]
fn test_det() {
for i in 5..21 {
let df = DataFrame::read_nc_by_header(
&format!("test_data/rand_mat/randmat_{}.nc", i),
vec!["m", "det"],
)
.unwrap();
let a_vec: Vec<f64> = df["m"].to_vec();
let a: Matrix = matrix(a_vec, i, i, Col);
let b: f64 = df["det"].at(0).unwrap();
let c = a.det();
assert!((b - c).abs() <= 1e-10);
}
}
#[cfg(feature = "nc")]
#[test]
fn test_pinv() {
for i in 5..21 {
let df = DataFrame::read_nc_by_header(
&format!("test_data/rand_mat/randmat_{}.nc", i),
vec!["pm", "pinv"],
)
.unwrap();
let a_vec: Vec<f64> = df["pm"].to_vec();
let b_vec: Vec<f64> = df["pinv"].to_vec();
let a: Matrix = matrix(a_vec, i + 1, i - 1, Col);
let b: Matrix = matrix(b_vec, i - 1, i + 1, Col);
let c = a.pseudo_inv();
assert_eq!(b, c);
}
}
#[cfg(feature = "nc")]
#[test]
fn test_solve() {
for i in 5..21 {
let df = DataFrame::read_nc_by_header(
&format!("test_data/rand_mat/randmat_{}.nc", i),
vec!["m", "v", "sol"],
)
.unwrap();
let a_vec: Vec<f64> = df["m"].to_vec();
let a: Matrix = matrix(a_vec, i, i, Col);
let b: Vec<f64> = df["v"].to_vec();
let x: Vec<f64> = df["sol"].to_vec();
let c = a.solve(&b, LU);
let d = a.solve(&b, WAZ);
x.print();
c.print();
d.print();
assert!(eq_vec(&x, &c, 1e-6));
assert!(eq_vec(&x, &d, 1e-6));
}
}