csaps 0.5.0

Cubic spline approximation (smoothing)
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

use csaps::CubicSmoothingSpline;
use ndarray::array;

#[test]
fn test_evaluate_1d() {
    let x = array![1., 2., 3., 4.];
    let y = array![1., 2., 3., 4.];
    let xi = array![1., 1.5, 2., 2.5, 3., 3.5, 4.];

    let spline = CubicSmoothingSpline::new(&x, &y).make().unwrap();

    let ys = spline.evaluate(&xi).unwrap();

    assert_eq!(ys, array![1., 1.5, 2., 2.5, 3., 3.5, 4.]);
}

#[test]
fn test_evaluate_2d_1() {
    let x = array![1., 2., 3., 4.];
    let y = array![[1., 2., 3., 4.], [3., 5., 7., 9.]];

    let spline = CubicSmoothingSpline::new(&x, &y).make().unwrap();

    let ys = spline.evaluate(&x).unwrap();

    assert_eq!(ys, y);
}

#[test]
fn test_evaluate_2d_2() {
    let x = array![1., 2., 3., 4.];
    let y = array![[1., 2., 3., 4.], [3., 5., 7., 9.]];
    let xi = array![1., 1.5, 2., 2.5, 3., 3.5, 4.];

    let spline = CubicSmoothingSpline::new(&x, &y).make().unwrap();

    let ys = spline.evaluate(&xi).unwrap();

    assert_eq!(
        ys,
        array![
            [1.0, 1.5, 2.0, 2.5, 3.0, 3.5, 4.0],
            [3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0]
        ]
    );
}

#[test]
fn test_evaluate_3d() {
    let x = array![1., 2., 3., 4.];
    let y = array![
        [[1., 2., 3., 4.], [2., 4., 6., 8.]],
        [[1., 3., 5., 7.], [3., 5., 7., 9.]]
    ];

    let spline = CubicSmoothingSpline::new(&x, &y).make().unwrap();

    let ys = spline.evaluate(&x).unwrap();

    assert_eq!(ys, y);
}