use polars_row_derive::IterToDataFrame;

#[derive(IterToDataFrame)]
pub struct TestStruct {
    a: i32,
    b: i32,
}

#[test]
fn works_exact() {
    let df = (0..10)
        .map(|i| TestStruct { a: i, b: i })
        .to_dataframe()
        .unwrap();
    assert_eq!(df.shape(), (10, 2));
}

#[test]
fn works_dyn() {
    let df = (0..10)
        .filter(|i| i % 2 == 0)
        .map(|i| TestStruct { a: i, b: i })
        .to_dataframe_dyn()
        .unwrap();
    assert_eq!(df.shape(), (5, 2));
}