use crate::eval::coercion::to_number;
use crate::eval::functions::check_arity;
use crate::types::{ErrorKind, Value};
pub fn power_fn(args: &[Value]) -> Value {
if let Some(err) = check_arity(args, 2, 2) {
return err;
}
let base = match to_number(args[0].clone()) {
Err(e) => return e,
Ok(v) => v,
};
let exp = match to_number(args[1].clone()) {
Err(e) => return e,
Ok(v) => v,
};
let result = if base < 0.0 && exp.fract() != 0.0 {
let inv = 1.0 / exp;
let inv_round = inv.round();
if (inv - inv_round).abs() < 1e-9 && (inv_round.abs() as i64) % 2 == 1 {
let mag = libm::pow(base.abs(), exp);
-mag
} else {
return Value::Error(ErrorKind::Num);
}
} else {
libm::pow(base, exp)
};
if !result.is_finite() {
return Value::Error(ErrorKind::Num);
}
Value::Number(result)
}
#[cfg(test)]
mod tests;