use ndarray::Array1;
pub fn happycat(x: &Array1<f64>) -> f64 {
let n = x.len() as f64;
let norm_sq: f64 = x.iter().map(|&xi| xi.powi(2)).sum();
let sum_x: f64 = x.iter().sum();
((norm_sq - n).abs()).powf(0.25) + (0.5 * norm_sq + sum_x) / n + 0.5
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_happycat_known_properties() {
use crate::{FunctionMetadata, get_function_metadata};
use ndarray::Array1;
let metadata = get_function_metadata();
let meta = metadata
.get("happycat")
.expect("Function happycat 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 = happycat(&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 = happycat(&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"
);
}
}
}