russell_sparse 1.12.0

Solvers for large sparse linear systems (wraps MUMPS and UMFPACK)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

use russell_lab::*;
use russell_sparse::prelude::*;

#[test]
fn test_complex_umfpack() -> Result<(), StrError> {
    let n = 10;
    let d = (n as f64) / 10.0;
    let mut coo = ComplexCooMatrix::new(n, n, n, Sym::No)?;
    let mut x_correct = ComplexVector::new(n);
    let mut rhs = ComplexVector::new(n);
    for k in 0..n {
        // put diagonal entries
        let akk = cpx!(10.0 + (k as f64) * d, 10.0 - (k as f64) * d);
        coo.put(k, k, akk)?;
        // let the exact solution be k + 0.5i
        x_correct[k] = cpx!(k as f64, 0.5);
        // generate RHS to match solution
        rhs[k] = akk * x_correct[k];
    }
    // println!("a =\n{}", coo.as_dense());
    // println!("x =\n{}", x_correct);
    // println!("b =\n{}", rhs);
    let mut x = ComplexVector::new(n);
    let mut solver = ComplexSolverUMFPACK::new()?;
    solver.factorize(&coo, None)?;
    solver.solve(&mut x, &rhs, false)?;
    complex_vec_approx_eq(&x, &x_correct, 1e-14);
    Ok(())
}