alopex-dataframe 0.7.3

Polars-compatible DataFrame API for Alopex DB (v0.1)
Documentation 
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

use std::sync::Arc;

use alopex_dataframe::{DataFrame, DataFrameError, Series};
use arrow::array::{ArrayRef, Int64Array};

fn s_i64(name: &str, values: Vec<i64>) -> Series {
    let array: ArrayRef = Arc::new(Int64Array::from(values));
    Series::from_arrow(name, vec![array]).unwrap()
}

#[cfg_attr(not(feature = "lane_ci"), ignore)]
#[test]
fn unique_keeps_first_occurrence() {
    let df = DataFrame::new(vec![
        s_i64("id", vec![1, 1, 2, 2]),
        s_i64("val", vec![10, 10, 20, 30]),
    ])
    .unwrap();

    let out = df.unique(None).unwrap();
    assert_eq!(out.height(), 3);

    let ids = out.column("id").unwrap().to_arrow();
    let ids = ids[0].as_any().downcast_ref::<Int64Array>().unwrap();
    assert_eq!(ids.value(0), 1);
    assert_eq!(ids.value(1), 2);
    assert_eq!(ids.value(2), 2);
}

#[cfg_attr(not(feature = "lane_ci"), ignore)]
#[test]
fn unique_subset_keeps_first_per_key() {
    let df = DataFrame::new(vec![
        s_i64("id", vec![1, 1, 2, 2]),
        s_i64("val", vec![10, 11, 20, 30]),
    ])
    .unwrap();

    let out = df.unique(Some(vec!["id".to_string()])).unwrap();
    assert_eq!(out.height(), 2);

    let vals = out.column("val").unwrap().to_arrow();
    let vals = vals[0].as_any().downcast_ref::<Int64Array>().unwrap();
    assert_eq!(vals.value(0), 10);
    assert_eq!(vals.value(1), 20);
}

#[cfg_attr(not(feature = "lane_ci"), ignore)]
#[test]
fn unique_errors_on_missing_subset_column() {
    let df = DataFrame::new(vec![s_i64("id", vec![1])]).unwrap();
    let err = df.unique(Some(vec!["missing".to_string()])).unwrap_err();
    assert!(matches!(err, DataFrameError::ColumnNotFound { .. }));
}