use polars_core::prelude::*;
use polars_core::series::ops::NullBehavior;

use crate::prelude::diff;

pub fn pct_change(s: &Series, n: &Series) -> PolarsResult<Series> {
    polars_ensure!(
        n.len() == 1,
        ComputeError: "n must be a single value."
    );

    match s.dtype() {
        DataType::Float64 | DataType::Float32 => {},
        _ => return pct_change(&s.cast(&DataType::Float64)?, n),
    }

    let fill_null_s = s.fill_null(FillNullStrategy::Forward(None))?;

    let n_s = n.cast(&DataType::Int64)?;
    if let Some(n) = n_s.i64()?.get(0) {
        diff(&fill_null_s, n, NullBehavior::Ignore)?.divide(&fill_null_s.shift(n))
    } else {
        Ok(Series::full_null(s.name().clone(), s.len(), s.dtype()))
    }
}
