Function check_window_size

pub fn check_window_size(
    df: &DataFrame,
    window: usize,
    indicator_name: &str,
) -> PolarsResult<()>

Check if a DataFrame has enough rows for a given window size

Arguments

• df - The DataFrame to check
• window - The window size required
• indicator_name - Name of the indicator (for error message)

Returns

Returns a PolarsResult<()> or an error if there are not enough rows

Example

use polars::prelude::*;
use ta_lib_in_rust::util::dataframe_utils::check_window_size;

let df = DataFrame::new(vec![Series::new("close", &[1.0, 2.0, 3.0, 4.0])]).unwrap();
assert!(check_window_size(&df, 3, "test").is_ok());
assert!(check_window_size(&df, 5, "test").is_err());