use super::*;
pub(crate) fn measure(
lf: &LazyFrame,
total_rows: Option<usize>,
plan: &DataQualityPlan,
) -> DataQualityResults {
compute_data_quality(lf, total_rows, plan, None, false).unwrap()
}
pub(crate) fn profile(name: &str, dtype: DataType) -> ColumnQualityProfile {
ColumnQualityProfile::unmeasured(name, dtype, 100)
}
pub(crate) fn observation(
kind: ObservationKind,
column: &str,
affected: usize,
) -> QualityObservation {
QualityObservation {
kind,
column: column.to_string(),
affected_rows: affected,
evaluated_rows: 100,
fact: String::new(),
normalized_category: None,
files: Vec::new(),
time_format: None,
full_scale: None,
}
}
pub(crate) fn results_with(
columns: Vec<ColumnQualityProfile>,
observations: Vec<QualityObservation>,
) -> DataQualityResults {
let mut results = DataQualityResults::empty(Some(100), &Schema::default());
results.evaluated_rows = 100;
results.precision = QualityPrecision::Exact;
results.columns = columns;
results.observations = observations;
results
}