ndarray-odeint 0.7.0

solve ODE using rust-ndarray
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
30
31
32
33
34
35
36
37
38
39
40
41
42


extern crate ndarray;
extern crate ndarray_odeint;
#[macro_use]
extern crate ndarray_linalg;

use ndarray::*;
use ndarray_linalg::*;
use ndarray_odeint::*;

#[test]
fn jacobian_linearity() {
    let eom = model::Lorenz63::default();
    let teo = explicit::rk4(eom, 0.01);
    let x0 = arr1(&[1.0, 0.0, 0.0]);
    let j = jacobian(&teo, x0, 1e-7);
    let v: Array1<f64> = generate::random(3);
    let w: Array1<f64> = generate::random(3);
    let jv_jw = j.op(&v) + j.op(&w);
    let j_vw = j.op_into(v + w);
    assert_close_l2!(&jv_jw, &j_vw, 1e-5);
}

#[test]
fn jacobian_view() {
    let eom = model::Lorenz63::default();
    let teo = explicit::rk4(eom, 0.01);
    let x0 = arr1(&[1.0, 0.0, 0.0]);
    let j = jacobian(&teo, x0, 1e-7);
    let mut v: Array1<f64> = generate::random(3);
    j.op_mut(&mut v.view_mut());
}

#[test]
fn jacobian_2d() {
    let eom = model::Lorenz63::default();
    let teo = explicit::rk4(eom, 0.01);
    let x0 = arr1(&[1.0, 0.0, 0.0]);
    let j = jacobian(&teo, x0, 1e-7);
    let v: Array2<f64> = generate::random((3, 2));
    j.op_multi(&v);
}