use ndarray::Array1;
pub fn sharp_ridge(x: &Array1<f64>) -> f64 {
if x.is_empty() {
return 0.0;
}
let x1 = x[0];
let sum_rest: f64 = x.iter().skip(1).map(|&xi| xi.powi(2)).sum();
x1.powi(2) + 100.0 * sum_rest.sqrt()
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_sharp_ridge_known_properties() {
use crate::{FunctionMetadata, get_function_metadata};
use ndarray::Array1;
let metadata = get_function_metadata();
let meta = metadata
.get("sharp_ridge")
.expect("Function sharp_ridge 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 = sharp_ridge(&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 = sharp_ridge(&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"
);
}
}
}