invr 0.2.3

Declarative data validation engine using invariants executed on Polars DataFrames.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

pub(crate) mod mean_between;
pub(crate) mod stddev_max;
pub(crate) mod sum_between;
pub(crate) mod value_between;
pub(crate) mod value_max;
pub(crate) mod value_min;

use crate::invariant::Invariant;
use crate::prelude::PolarsKind;
use crate::violation::Violation;
use polars::datatypes::AnyValue;

pub fn map_count_violation(
    inv: &Invariant<PolarsKind>,
    value: AnyValue,
    label: &str,
) -> Option<Violation> {
    let count = value.try_extract::<i64>().ok()?;

    if count > 0 {
        Some(Violation::new(
            inv.id().clone(),
            inv.severity(),
            inv.scope().clone(),
            format!("{count} violations of {label}"),
        ))
    } else {
        None
    }
}