nalgebra/tests/core/blas.rs

#![cfg(feature = "arbitrary")]

use std::cmp;
use na::{DVector, DMatrix};

quickcheck! {
    /*
     *
     * Symmetric operators.
     *
     */
    fn gemv_symm(n: usize, alpha: f64, beta: f64) -> bool {
        let n = cmp::max(1, cmp::min(n, 50));
        let a = DMatrix::<f64>::new_random(n, n);
        let a = &a * a.transpose();

        let x = DVector::new_random(n);
        let mut y1 = DVector::new_random(n);
        let mut y2 = y1.clone();

        y1.gemv(alpha, &a, &x, beta);
        y2.gemv_symm(alpha, &a.lower_triangle(), &x, beta);

        if !relative_eq!(y1, y2, epsilon = 1.0e-10) {
            return false;
        }

        y1.gemv(alpha, &a, &x, 0.0);
        y2.gemv_symm(alpha, &a.lower_triangle(), &x, 0.0);

        relative_eq!(y1, y2, epsilon = 1.0e-10)
    }

    fn ger_symm(n: usize, alpha: f64, beta: f64) -> bool {
        let n = cmp::max(1, cmp::min(n, 50));
        let a = DMatrix::<f64>::new_random(n, n);
        let mut a1 = &a * a.transpose();
        let mut a2 = a1.lower_triangle();

        let x = DVector::new_random(n);
        let y = DVector::new_random(n);

        a1.ger(alpha, &x, &y, beta);
        a2.ger_symm(alpha, &x, &y, beta);

        if !relative_eq!(a1.lower_triangle(), a2) {
            return false;
        }

        a1.ger(alpha, &x, &y, 0.0);
        a2.ger_symm(alpha, &x, &y, 0.0);

        relative_eq!(a1.lower_triangle(), a2)
    }

    fn quadform_symm(n: usize, alpha: f64, beta: f64) -> bool {
        let n = cmp::max(1, cmp::min(n, 50));
        let lhs         = DMatrix::<f64>::new_random(6, n);
        let mut mid     = DMatrix::<f64>::new_random(n, n);
        let mut res     = DMatrix::new_random(6, 6);
        let mut scratch = Vector6::zeros();

        mid.fill_upper_triangle_with_lower_triangle();

        let expected = &res * beta + &lhs * &mid * lhs.transpose() * alpha;

        res.quadform_symm(&mut scratch, alpha, &lhs, &mid, beta);
        res.fill_upper_triangle_with_lower_triangle();

        println!("{}{}", res, expected);

        relative_eq!(res.lower_triangle(), expected.lower_triangle(), epsilon = 1.0e-7)
    }
}