conspire 0.6.0

The Rust interface to conspire.
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

use crate::{
    EPSILON,
    math::{
        Scalar,
        test::{TestError, assert_eq_from_fd},
    },
    physics::molecular::potential::{Harmonic, Morse, Potential},
};

const NUM: usize = 333;

#[test]
fn finite_difference() -> Result<(), TestError> {
    let e = 1.2;
    let a = 1.1;
    let x0 = 1.5;
    let x_max = x0 + 0.98 * 2.0_f64.ln() / a;
    let t = 1e-1;
    let potential = Harmonic {
        rest_length: x0,
        stiffness: e,
    };
    (0..NUM)
        .map(|k| x0 + (x_max - x0) * k as Scalar / NUM as Scalar)
        .into_iter()
        .try_for_each(|mut x| {
            let mut force = potential.force(x);
            let stiffness = potential.stiffness(x);
            let anharmonicity = potential.anharmonicity(x);
            x += 0.5 * EPSILON;
            let mut force_fd = potential.energy(x);
            let mut stiffness_fd = potential.force(x);
            let mut anharmonicity_fd = potential.stiffness(x);
            x -= EPSILON;
            force_fd = (force_fd - potential.energy(x)) / EPSILON;
            stiffness_fd = (stiffness_fd - potential.force(x)) / EPSILON;
            anharmonicity_fd = (anharmonicity_fd - potential.stiffness(x)) / EPSILON;
            assert_eq_from_fd(&force, &force_fd)?;
            assert_eq_from_fd(&stiffness, &stiffness_fd)?;
            assert_eq_from_fd(&anharmonicity, &anharmonicity_fd)?;
            let extension = potential.extension(force);
            let compliance = potential.compliance(force);
            force += 0.5 * EPSILON;
            let mut extension_fd = potential.legendre(force);
            let mut compliance_fd = potential.extension(force);
            force -= EPSILON;
            extension_fd = (potential.legendre(force) - extension_fd) / EPSILON;
            compliance_fd = (compliance_fd - potential.extension(force)) / EPSILON;
            assert_eq_from_fd(&extension, &extension_fd)?;
            assert_eq_from_fd(&compliance, &compliance_fd)
        })?;
    let potential = Morse {
        rest_length: x0,
        depth: e,
        parameter: a,
    };
    (1..NUM)
        .map(|k| x0 + (x_max - x0) * k as Scalar / NUM as Scalar)
        .into_iter()
        .try_for_each(|mut x| {
            let mut force = potential.force(x);
            let stiffness = potential.stiffness(x);
            let anharmonicity = potential.anharmonicity(x);
            x += 0.5 * EPSILON;
            let mut force_fd = potential.energy(x);
            let mut stiffness_fd = potential.force(x);
            let mut anharmonicity_fd = potential.stiffness(x);
            x -= EPSILON;
            force_fd = (force_fd - potential.energy(x)) / EPSILON;
            stiffness_fd = (stiffness_fd - potential.force(x)) / EPSILON;
            anharmonicity_fd = (anharmonicity_fd - potential.stiffness(x)) / EPSILON;
            assert_eq_from_fd(&force, &force_fd)?;
            assert_eq_from_fd(&stiffness, &stiffness_fd)?;
            assert_eq_from_fd(&anharmonicity, &anharmonicity_fd)?;
            let extension = potential.extension(force);
            let compliance = potential.compliance(force);
            let nondimensional_extension = potential.nondimensional_extension(force, t);
            force += 0.5 * EPSILON;
            let mut extension_fd = potential.legendre(force);
            let mut compliance_fd = potential.extension(force);
            let mut nondimensional_extension_fd = potential.nondimensional_legendre(force, t);
            force -= EPSILON;
            extension_fd = (potential.legendre(force) - extension_fd) / EPSILON;
            compliance_fd = (compliance_fd - potential.extension(force)) / EPSILON;
            nondimensional_extension_fd = (potential.nondimensional_legendre(force, t)
                - nondimensional_extension_fd)
                / EPSILON;
            assert_eq_from_fd(&extension, &extension_fd)?;
            assert_eq_from_fd(&compliance, &compliance_fd)?;
            assert_eq_from_fd(&nondimensional_extension, &nondimensional_extension_fd)
        })
}