46 lines
1.0 KiB
Rust
46 lines
1.0 KiB
Rust
|
#[path = "../benches/test_functions.rs"]
|
||
|
|
mod test_functions;
|
||
|
|
|
||
|
|
use test_functions::*;
|
||
|
|
|
||
|
|
const TOL: f64 = 1e-10;
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn sphere_at_optimum() {
|
||
|
|
assert!(sphere(&[0.0, 0.0]).abs() < TOL);
|
||
|
|
assert!(sphere(&[0.0; 10]).abs() < TOL);
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn rosenbrock_at_optimum() {
|
||
|
|
assert!(rosenbrock(&[1.0, 1.0]).abs() < TOL);
|
||
|
|
assert!(rosenbrock(&[1.0; 5]).abs() < TOL);
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn rastrigin_at_optimum() {
|
||
|
|
assert!(rastrigin(&[0.0, 0.0]).abs() < TOL);
|
||
|
|
assert!(rastrigin(&[0.0; 10]).abs() < TOL);
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn ackley_at_optimum() {
|
||
|
|
assert!(ackley(&[0.0, 0.0]).abs() < 1e-8);
|
||
|
|
assert!(ackley(&[0.0; 10]).abs() < 1e-8);
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn branin_at_optimum() {
|
||
|
|
let target = 0.397_887_357_729_738_1;
|
||
|
|
let val = branin(&[std::f64::consts::PI, 2.275]);
|
||
|
|
assert!((val - target).abs() < 1e-3);
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn hartmann6_at_optimum() {
|
||
|
|
let target = -3.3224;
|
||
|
|
let x_opt = [0.20169, 0.150011, 0.476874, 0.275332, 0.311652, 0.6573];
|
||
|
|
let val = hartmann6(&x_opt);
|
||
|
|
assert!((val - target).abs() < 0.01);
|
||
|
|
}
|