use ndarray::Array1;
pub fn gramacy_lee_2012(x: &Array1<f64>) -> f64 {
let x1 = x[0];
let pi = std::f64::consts::PI;
if x1 == 0.0 {
5.0 * pi + (x1 - 1.0).powi(4)
} else {
(10.0 * pi * x1).sin() / (2.0 * x1) + (x1 - 1.0).powi(4)
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_gramacy_lee_2012_known_properties() {
use crate::{FunctionMetadata, get_function_metadata};
use ndarray::Array1;
let metadata = get_function_metadata();
let meta = metadata
.get("gramacy_lee_2012")
.expect("Function gramacy_lee_2012 should have metadata");
for (minimum_coords, expected_value) in &meta.global_minima {
assert!(
minimum_coords.len() >= meta.bounds.len() || meta.bounds.len() == 1,
"Global minimum coordinates should match bounds dimensions"
);
for (i, &coord) in minimum_coords.iter().enumerate() {
if i < meta.bounds.len() {
let (lower, upper) = meta.bounds[i];
assert!(
coord >= lower && coord <= upper,
"Global minimum coordinate {} = {} should be within bounds [{} {}]",
i,
coord,
lower,
upper
);
}
}
}
let tolerance = 1e-6; for (minimum_coords, expected_value) in &meta.global_minima {
let x = Array1::from_vec(minimum_coords.clone());
let actual_value = gramacy_lee_2012(&x);
let error = (actual_value - expected_value).abs();
assert!(
error <= tolerance,
"Function value at global minimum {:?} should be {}, got {}, error: {}",
minimum_coords,
expected_value,
actual_value,
error
);
}
if !meta.global_minima.is_empty() {
let (first_minimum, _) = &meta.global_minima[0];
let x = Array1::from_vec(first_minimum.clone());
let result = gramacy_lee_2012(&x);
assert!(
result.is_finite(),
"Function should return finite values at global minimum"
);
assert!(
!result.is_nan(),
"Function should not return NaN at global minimum"
);
}
}
}