ginger-rs 0.1.7

#![allow(non_snake_case)]

use super::horner::{horner_eval_c, horner_eval_f};
use super::Options;
use crate::execution_policy::{atomic_decoupled_run, jacobi_mt_run, sequential_run, Step};
use crate::leja_order::leja_order;
use crate::scaling::{derivative_coeffs, scaled_coeffs};
use crate::seqlock::AtomicComplex;
use num_complex::Complex;

const TWO_PI: f64 = std::f64::consts::TAU;

/// Initial guess for Aberth's method using low-discrepancy sequence
///
/// The center $$ c $$ and radius $$ R $$ are derived from the coefficients:
///
/// $$ c = -\frac{a_1}{n a_0}, \qquad R = \sqrt\[n\]{-P(c)}, \qquad z_i = c + R \cdot (\cos\theta_i + i\sin\theta_i) $$
///
/// where $$ \theta_i = 2\pi v_i $$ with $$ v_i $$ from a van der Corput low-discrepancy sequence.
pub fn initial_aberth(coeffs: &[f64]) -> Vec<Complex<f64>> {
    let degree = coeffs.len() - 1;
    let center = -coeffs[1] / (coeffs[0] * degree as f64);
    let poly_c = horner_eval_f(coeffs, center);
    let radius = Complex::<f64>::new(-poly_c, 0.0).powf(1.0 / degree as f64);
    (0..degree)
        .map(|i| {
            // note! swap x and y to match C++ Complex{circle2_table_y, circle2_table_x}
            let xcoord = crate::tables::circle2_table_y(i);
            let ycoord = crate::tables::circle2_table_x(i);
            center + radius * Complex::<f64>::new(xcoord, ycoord)
        })
        .collect()
}

/// Initial guess for Aberth's method
///
/// The center $$ c $$ and radius $$ R $$ are derived from the coefficients:
///
/// $$ c = -\frac{a_1}{n a_0}, \qquad R = \sqrt\[n\]{-P(c)}, \qquad z_i = c + R \cdot e^{i\theta_i} $$
///
/// where $$ \theta_i = 2\pi v_i $$ with $$ v_i $$ from a van der Corput sequence
/// for low-discrepancy angular spacing.
///
/// The `initial_aberth` function calculates the initial guesses for Aberth's method given a
/// polynomial's coefficients.
///
/// Arguments:
///
/// * `coeffs`: The `coeffs` parameter is a slice of `f64` values representing the coefficients of a
///   polynomial. The coefficients are ordered from highest degree to lowest degree. For example, if the
///   polynomial is `3x^2 + 2x + 1`, the `coeffs` slice would
///
/// Returns:
///
/// The function `initial_aberth` returns a vector of `Complex<f64>` values, which represent the initial
/// guesses for the roots of a polynomial.
///
/// # Examples:
///
/// ```
/// use ginger::aberth::initial_aberth_orig;
/// use num_complex::Complex;
/// use approx_eq::assert_approx_eq;
///
/// let coeffs = vec![10.0, 34.0, 75.0, 94.0, 150.0, 94.0, 75.0, 34.0, 10.0];
/// let z0s = initial_aberth_orig(&coeffs);
///
/// assert_approx_eq!(z0s[0].re, 0.6116610247366323);
/// assert_approx_eq!(z0s[0].im, 0.6926747514925476);
/// ```
/// Original initial guess for Aberth's method (equi-angular spacing)
///
/// $$ z_i = c + R \cdot \left(\cos\frac{2\pi k}{n} + i\sin\frac{2\pi k}{n}\right) $$
pub fn initial_aberth_orig(coeffs: &[f64]) -> Vec<Complex<f64>> {
    let degree = coeffs.len() - 1;
    let center = -coeffs[1] / (coeffs[0] * degree as f64);
    let poly_c = horner_eval_f(coeffs, center);
    let radius = Complex::<f64>::new(-poly_c, 0.0).powf(1.0 / degree as f64);
    let k = TWO_PI / (degree as f64);
    (0..degree)
        .map(|idx| {
            let theta = k * (0.25 + idx as f64);
            center + radius * Complex::<f64>::new(theta.cos(), theta.sin())
        })
        .collect()
}

/// One Aberth-Ehrlich correction for a single root.
///
/// This mirrors `ginger::detail::aberth_step` in ginger-cpp.
pub struct AberthStep<'a> {
    /// Polynomial coefficients (highest degree first).
    pub coeffs: &'a [f64],
    /// Derivative coefficients of `coeffs`.
    pub coeffs1: &'a [f64],
}

impl Step<Complex<f64>> for AberthStep<'_> {
    type Cell = AtomicComplex;

    fn run<G, N>(&self, idx: usize, get: G, neighbors: N) -> (f64, Option<Complex<f64>>)
    where
        G: Fn(usize) -> Complex<f64>,
        N: Iterator<Item = usize>,
    {
        let zi = get(idx);
        let p_eval = horner_eval_c(self.coeffs, &zi);
        let tol_i = p_eval.l1_norm(); // ???
        let mut p1_eval = horner_eval_c(self.coeffs1, &zi);
        for j in neighbors {
            let zj = get(j);
            p1_eval -= p_eval / (zi - zj);
        }
        (tol_i, Some(zi - p_eval / p1_eval)) // Gauss-Seidel fashion
    }
}

/// Aberth's method
///
/// The `aberth` function implements Aberth's method for finding roots of a polynomial.
///
/// Each estimate $$ z_i $$ is updated by the correction:
///
/// $$ z_i' = z_i - \frac{P(z_i)}{P'(z_i)} $$
///
/// where the derivative is modified by all other root estimates:
///
/// $$ P'(z_i) = P_1(z_i) - \sum_{\substack{j=1\\j\neq i}}^n \frac{P(z_i)}{z_i - z_j} $$
///
/// Here $$ P_1(z) $$ is the ordinary derivative of $$ P(z) $$. Convergence is cubic.
///
/// Arguments:
///
/// * `coeffs`: The `coeffs` parameter is a slice of `f64` values representing the coefficients of a
///   polynomial. The coefficients are ordered from highest degree to lowest degree. For example, if the
///   polynomial is `3x^2 + 2x + 1`, the `coeffs` slice would
/// * `zs`: A vector of complex numbers representing the initial guesses for the roots of the polynomial.
/// * `options`: The `options` parameter is an instance of the `Options` struct, which contains the
///   following fields:
///
/// # Examples:
///
/// ```
/// use ginger::rootfinding::Options;
/// use ginger::aberth::{initial_aberth, aberth};
///
/// let coeffs = vec![10.0, 34.0, 75.0, 94.0, 150.0, 94.0, 75.0, 34.0, 10.0];
/// let mut zrs = initial_aberth(&coeffs);
/// let (niter, _found) = aberth(&coeffs, &mut zrs, &Options::default());
///
/// assert_eq!(niter, 5);
/// ```
#[cfg_attr(feature = "doc-images", doc = svgbobdoc::transform!(
/// ```svgbob
///  .───────────.    .───────────.    .──────────────.
///  │ Polynomial│───►│  Initial  │───►│  Aberth      │
///  │ coeffs    │    │  Guess    │    │  Iteration   │
///  '───────────'    '───────────'    '──────┬───────'
///                                           │
///                                      .────▼───────.
///                                      │ Converged?  │
///                                      '────┬───────'
///                                     No   │   Yes
///                                .─────────┘    │
///                                ▼              ▼
///                           .───────────.  .───────────.
///                           │ Iterate   │  │  Roots    │
///                           '───────────'  '───────────'
/// ```
))]
pub fn aberth(coeffs: &[f64], zs: &mut [Complex<f64>], options: &Options) -> (usize, bool) {
    let scaled = scaled_coeffs(coeffs);
    let coeffs1 = derivative_coeffs(&scaled);
    let step = AberthStep {
        coeffs: &scaled,
        coeffs1: &coeffs1,
    };
    sequential_run(zs, options, &step)
}

/// Multi-threading Aberth's method
///
/// Each estimate $$ z_i $$ is updated by the correction:
///
/// $$ z_i' = z_i - \frac{P(z_i)}{P'(z_i)} $$
///
/// where the derivative is modified by all other root estimates:
///
/// $$ P'(z_i) = P_1(z_i) - \sum_{\substack{j=1\\j\neq i}}^n \frac{P(z_i)}{z_i - z_j} $$
///
/// The `aberth_mt` function in Rust implements the multi-threaded Aberth's method for root finding.
///
/// Arguments:
///
/// * `coeffs`: The `coeffs` parameter is a slice of `f64` values representing the coefficients of a
///   polynomial. The polynomial is defined by the equation:
/// * `zs`: A mutable reference to a vector of Complex numbers. These numbers represent the initial
///   guesses for the roots of the polynomial equation.
/// * `options`: The `options` parameter is an instance of the `Options` struct, which contains the
///   following fields:
///
/// # Examples:
///
/// ```
/// use ginger::rootfinding::Options;
/// use ginger::aberth::{initial_aberth, aberth_mt};
///
/// let coeffs = vec![10.0, 34.0, 75.0, 94.0, 150.0, 94.0, 75.0, 34.0, 10.0];
/// let mut zrs = initial_aberth(&coeffs);
/// let (niter, _found) = aberth_mt(&coeffs, &mut zrs, &Options::default());
///
/// assert_eq!(niter, 6);
/// ```
pub fn aberth_mt(coeffs: &[f64], zs: &mut [Complex<f64>], options: &Options) -> (usize, bool) {
    let scaled = scaled_coeffs(coeffs);
    let coeffs1 = derivative_coeffs(&scaled);
    let step = AberthStep {
        coeffs: &scaled,
        coeffs1: &coeffs1,
    };
    jacobi_mt_run(zs, options, &step)
}

/// Atomic multi-threading Aberth's method (decoupled)
///
/// Each estimate $$ z_i $$ is updated by the correction:
///
/// $$ z_i' = z_i - \frac{P(z_i)}{P'(z_i)} $$
///
/// where the derivative is modified by all other root estimates:
///
/// $$ P'(z_i) = P_1(z_i) - \sum_{\substack{j=1\\j\neq i}}^n \frac{P(z_i)}{z_i - z_j} $$
///
/// Unlike `aberth_mt` (Jacobi snapshot + per-iteration barrier), this variant
/// builds a single atomic working buffer once: each thread owns a chunk of root
/// slots (single-writer, multi-reader via a seqlock) and runs its own iteration
/// loop independently. There is no per-iteration synchronization — a thread
/// exits as soon as its own chunk converges or the maximum number of
/// iterations is exceeded. The returned iteration count is the maximum across
/// threads, and `found` is true only if every chunk converged.
///
/// Arguments:
///
/// * `coeffs`: The `coeffs` parameter is a slice of `f64` values representing the coefficients of a
///   polynomial. The polynomial is defined by the equation:
/// * `zs`: A mutable reference to a vector of Complex numbers. These numbers represent the initial
///   guesses for the roots of the polynomial equation.
/// * `options`: The `options` parameter is an instance of the `Options` struct, which contains the
///   following fields:
///
/// # Examples:
///
/// ```
/// use ginger::rootfinding::Options;
/// use ginger::aberth::{initial_aberth, aberth_atomic};
///
/// let coeffs = vec![10.0, 34.0, 75.0, 94.0, 150.0, 94.0, 75.0, 34.0, 10.0];
/// let mut zrs = initial_aberth(&coeffs);
/// let (niter, _found) = aberth_atomic(&coeffs, &mut zrs, &Options::default());
///
/// assert!(niter > 0);
/// ```
pub fn aberth_atomic(coeffs: &[f64], zs: &mut [Complex<f64>], options: &Options) -> (usize, bool) {
    let scaled = scaled_coeffs(coeffs);
    let coeffs1 = derivative_coeffs(&scaled);
    let step = AberthStep {
        coeffs: &scaled,
        coeffs1: &coeffs1,
    };
    atomic_decoupled_run(zs, options, &step)
}

/// Initial guess for Aberth's method using auto-correlation
///
/// $$ R = \sqrt\[n\]{|a_n|},\qquad R \leftarrow \max(R, 1/R),\qquad z_i = c + R \cdot e^{i\theta_i} $$
///
/// The `initial_aberth_autocorr` function calculates initial guesses for Aberth's method
/// specifically tailored for auto-correlation polynomials.
///
/// Arguments:
///
/// * `coeffs`: The `coeffs` parameter is a slice of `f64` values representing the coefficients
///   of a polynomial. The coefficients are ordered from highest degree to lowest degree.
///
/// Returns:
///
/// The function returns a vector of `Complex<f64>` values, representing the initial guesses
/// for the roots of a polynomial.
pub fn initial_aberth_autocorr(coeffs: &[f64]) -> Vec<Complex<f64>> {
    let degree = coeffs.len() - 1; // assume even
    let center = -coeffs[1] / (coeffs[0] * degree as f64);
    let poly_c = horner_eval_f(coeffs, center);
    let mut radius = poly_c.abs().powf(1.0 / degree as f64);
    if radius > 1.0 {
        radius = 1.0 / radius;
    }
    (0..degree / 2)
        .map(|i| {
            // note! swap x and y to match C++ Complex{circle2_table_y, circle2_table_x}
            let xcoord = crate::tables::circle2_table_y(i);
            let ycoord = crate::tables::circle2_table_x(i);
            center + radius * Complex::<f64>::new(xcoord, ycoord)
        })
        .collect()
}

/// One Aberth-Ehrlich correction for a palindromic (auto-correlation) root;
/// each neighbor root contributes both `zj` and its reciprocal `1/zj`.
///
/// This mirrors `ginger::detail::aberth_autocorr_step` in ginger-cpp.
pub struct AberthAutocorrStep<'a> {
    /// Polynomial coefficients (highest degree first).
    pub coeffs: &'a [f64],
    /// Derivative coefficients of `coeffs`.
    pub coeffs1: &'a [f64],
}

impl Step<Complex<f64>> for AberthAutocorrStep<'_> {
    type Cell = AtomicComplex;

    fn run<G, N>(&self, idx: usize, get: G, neighbors: N) -> (f64, Option<Complex<f64>>)
    where
        G: Fn(usize) -> Complex<f64>,
        N: Iterator<Item = usize>,
    {
        let zi = get(idx);
        let p_eval = horner_eval_c(self.coeffs, &zi);
        let tol_i = p_eval.l1_norm(); // ???
        let mut p1_eval = horner_eval_c(self.coeffs1, &zi);
        for j in neighbors {
            let zj = get(j);
            p1_eval -= p_eval / (zi - zj);
            p1_eval -= p_eval / (zi - 1.0 / zj);
        }
        (tol_i, Some(zi - p_eval / p1_eval)) // Gauss-Seidel fashion
    }
}

/// Aberth's method for auto-correlation polynomials
///
/// Each estimate $$ z_i $$ is updated by the correction, considering both $$ z $$
/// and $$ 1/\bar{z} $$ in the derivative:
///
/// $$ P'(z_i) = P_1(z_i) - \sum_{\substack{j=1\\j\neq i}}^n
///    \left(\frac{P(z_i)}{z_i - z_j} + \frac{P(z_i)}{z_i - 1/z_j}\right) $$
///
/// The `aberth_autocorr` function implements Aberth's method specifically for
/// auto-correlation polynomials, where roots come in reciprocal pairs.
///
/// Arguments:
///
/// * `coeffs`: The polynomial coefficients (highest to lowest degree)
/// * `zs`: Mutable slice of complex root approximations
/// * `options`: Iteration options (max iterations, tolerance)
///
/// Returns:
///
/// A tuple of (number of iterations, whether convergence was achieved)
pub fn aberth_autocorr(
    coeffs: &[f64],
    zs: &mut [Complex<f64>],
    options: &Options,
) -> (usize, bool) {
    let scaled = scaled_coeffs(coeffs);
    let coeffs1 = derivative_coeffs(&scaled);
    let step = AberthAutocorrStep {
        coeffs: &scaled,
        coeffs1: &coeffs1,
    };
    sequential_run(zs, options, &step)
}

/// Reconstruct a monic polynomial from its roots using Leja ordering
///
/// $$ P(x) = \prod_{i=1}^n (x - r_i) = x^n + a_{n-1} x^{n-1} + \cdots + a_0 $$
///
/// The coefficients are computed by repeated convolution. Starting from $$ \[1\] $$,
/// for each root $$ r $$:
///
/// $$ c_{k+1} \leftarrow c_{k+1} - r \cdot c_k $$
///
/// Given a set of complex roots, reconstruct the monic polynomial coefficients
/// (highest degree first) by multiplying (x - root) factors. Leja ordering is
/// applied for numerical accuracy.
///
/// Arguments:
///
/// * `zs` - Input vector of complex roots
///
/// Returns:
///
/// Monic polynomial coefficients (highest degree first)
pub fn poly_from_roots(zs: &[Complex<f64>]) -> Vec<f64> {
    if zs.is_empty() {
        return vec![1.0];
    }
    let ordered = leja_order(zs.to_vec());
    let mut coeffs = vec![Complex::new(1.0, 0.0)];
    for z in &ordered {
        let mut prev = coeffs[0];
        for item in coeffs.iter_mut().skip(1) {
            let old = *item;
            *item -= z * prev;
            prev = old;
        }
        coeffs.push(-z * prev);
    }
    coeffs.iter().map(|c| c.re).collect()
}

/// Reconstruct a monic polynomial from its autocorrelation roots
///
/// Auto-correlation (palindromic) polynomials have roots in reciprocal pairs.
/// The `aberth_autocorr` functions find the degree/2 "independent" roots.
/// This function adds the reciprocal of each root $1/z$ to get the full set
/// of degree roots, then reconstructs with Leja ordering.
///
/// $$ P(x) = \prod_{i=1}^{n/2} (x - z_i)(x - 1/z_i) $$
///
/// Arguments:
///
/// * `zs` - Roots found by aberth_autocorr
///
/// Returns:
///
/// Monic polynomial coefficients (highest degree first)
pub fn poly_from_autocorr_roots(zs: &[Complex<f64>]) -> Vec<f64> {
    if zs.is_empty() {
        return vec![1.0];
    }
    // Add reciprocals to account for the palindromic root-pair structure
    let mut all_roots: Vec<Complex<f64>> = Vec::with_capacity(2 * zs.len());
    for z in zs {
        all_roots.push(*z);
        all_roots.push(1.0 / z);
    }
    poly_from_roots(&all_roots)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_horner_eval() {
        let coeffs = vec![10.0, 34.0, 75.0, 94.0, 150.0, 94.0, 75.0, 34.0, 10.0];
        let z = Complex::new(0.0, 0.0);
        let p_eval = horner_eval_c(&coeffs, &z);
        assert_eq!(p_eval.re, 10.0);
        assert_eq!(p_eval.im, 0.0);
        let z = Complex::new(1.0, 0.0);
        let p_eval = horner_eval_c(&coeffs, &z);
        assert_eq!(p_eval.re, 576.0);
        assert_eq!(p_eval.im, 0.0);
    }

    #[test]
    fn test_aberth() {
        let coeffs = vec![10.0, 34.0, 75.0, 94.0, 150.0, 94.0, 75.0, 34.0, 10.0];
        let mut zrs = initial_aberth(&coeffs);
        let (niter, found) = aberth(&coeffs, &mut zrs, &Options::default());
        assert_eq!(niter, 5);
        assert!(found);
    }

    #[test]
    fn test_aberth_mt() {
        let coeffs = vec![10.0, 34.0, 75.0, 94.0, 150.0, 94.0, 75.0, 34.0, 10.0];
        let mut zrs = initial_aberth(&coeffs);
        let (niter, found) = aberth_mt(&coeffs, &mut zrs, &Options::default());
        assert_eq!(niter, 6);
        assert!(found);
    }

    #[test]
    fn test_aberth_atomic() {
        let coeffs = vec![10.0, 34.0, 75.0, 94.0, 150.0, 94.0, 75.0, 34.0, 10.0];
        let mut zrs = initial_aberth(&coeffs);
        let (niter, found) = aberth_atomic(&coeffs, &mut zrs, &Options::default());
        assert!(found);
        assert!(niter <= 14);
    }

    const FIR_COEFFS: [f64; 49] = [
        -0.00196191,
        -0.00094597,
        -0.00023823,
        0.00134667,
        0.00380494,
        0.00681596,
        0.0097864,
        0.01186197,
        0.0121238,
        0.00985211,
        0.00474894,
        -0.00281751,
        -0.01173923,
        -0.0201885,
        -0.02590168,
        -0.02658216,
        -0.02035729,
        -0.00628271,
        0.01534627,
        0.04279982,
        0.0732094,
        0.10275561,
        0.12753013,
        0.14399228,
        0.15265722,
        0.14399228,
        0.12753013,
        0.10275561,
        0.0732094,
        0.04279982,
        0.01534627,
        -0.00628271,
        -0.02035729,
        -0.02658216,
        -0.02590168,
        -0.0201885,
        -0.01173923,
        -0.00281751,
        0.00474894,
        0.00985211,
        0.0121238,
        0.01186197,
        0.0097864,
        0.00681596,
        0.00380494,
        0.00134667,
        -0.00023823,
        -0.00094597,
        -0.00196191,
    ];

    #[test]
    fn test_aberth_atomic_fir() {
        let options = Options {
            tolerance: 1e-8,
            ..Options::default()
        };
        let mut zrs = initial_aberth(&FIR_COEFFS);
        let (niter, found) = aberth_atomic(&FIR_COEFFS, &mut zrs, &options);
        assert!(found);
        assert!(niter <= 100, "niter={niter}");
    }

    #[test]
    fn test_aberth_atomic_reconstruction() {
        let coeffs = vec![10.0, 34.0, 75.0, 94.0, 150.0, 94.0, 75.0, 34.0, 10.0];
        let mut zrs = initial_aberth(&coeffs);
        let (_, found) = aberth_atomic(&coeffs, &mut zrs, &Options::default());
        assert!(found);
        let monic = poly_from_roots(&zrs);
        let scale = coeffs[0];
        for (i, c) in coeffs.iter().enumerate() {
            assert!(
                (monic[i] * scale - c).abs() < 1e-8,
                "coefficient {i} mismatch: {} vs {}",
                monic[i] * scale,
                c
            );
        }
    }

    #[test]
    fn test_aberth_autocorr() {
        let coeffs = vec![10.0, 34.0, 75.0, 94.0, 150.0, 94.0, 75.0, 34.0, 10.0];
        let mut zrs = initial_aberth_autocorr(&coeffs);
        let (niter, found) = aberth_autocorr(&coeffs, &mut zrs, &Options::default());
        assert!(niter <= 7);
        assert!(found);
    }

    #[test]
    fn test_aberth_scaled_polynomial_converges() {
        // Same roots as the degree-8 palindromic polynomial, but with
        // coefficients large enough that an absolute residual test can never
        // reach the default 1e-12 tolerance.
        let base = vec![10.0, 34.0, 75.0, 94.0, 150.0, 94.0, 75.0, 34.0, 10.0];
        let scaled: Vec<f64> = base.iter().map(|c| c * 1e6).collect();
        let mut zs = initial_aberth(&scaled);
        let (niter, found) = aberth(&scaled, &mut zs, &Options::default());
        assert!(found);
        assert!(niter <= 12, "niter={niter}");
    }

    #[test]
    fn test_scaling_does_not_change_roots() {
        let base = vec![10.0, 34.0, 75.0, 94.0, 150.0, 94.0, 75.0, 34.0, 10.0];
        let scaled: Vec<f64> = base.iter().map(|c| c * 1e6).collect();
        let z0 = initial_aberth(&base);
        let mut za = z0.clone();
        let mut zb = z0.clone();
        let (na, oka) = aberth(&base, &mut za, &Options::default());
        let (nb, okb) = aberth(&scaled, &mut zb, &Options::default());
        assert!(oka && okb);
        assert_eq!(na, nb);
        for (a, b) in za.iter().zip(zb.iter()) {
            assert!((a - b).norm() < 1e-9, "{a} vs {b}");
        }
    }

    #[test]
    fn test_poly_from_roots() {
        // Polynomial (x-1)(x-2) = x^2 - 3x + 2
        let roots = vec![Complex::new(1.0, 0.0), Complex::new(2.0, 0.0)];
        let coeffs = poly_from_roots(&roots);
        assert_eq!(coeffs.len(), 3);
        assert!((coeffs[0] - 1.0).abs() < 1e-12);
        assert!((coeffs[1] + 3.0).abs() < 1e-12);
        assert!((coeffs[2] - 2.0).abs() < 1e-12);
    }

    #[test]
    fn test_poly_from_autocorr_roots() {
        // Simple test: just check it doesn't panic and returns right length
        let roots = vec![Complex::new(0.5, 0.5), Complex::new(0.5, -0.5)];
        let coeffs = poly_from_autocorr_roots(&roots);
        // Should have 2*roots.len() + 1 coefficients
        assert_eq!(coeffs.len(), 5);
        // Monic
        assert!((coeffs[0] - 1.0).abs() < 1e-12);
    }

    #[test]
    fn test_poly_from_roots_empty() {
        let coeffs = poly_from_roots(&[]);
        assert_eq!(coeffs, vec![1.0]);
    }
}