polars_rows_iter/iter_from_column/
iter_from_column_primitives.rs

1use super::*;
2use iter_from_column_trait::IterFromColumn;
3use polars_rows_iter_derive::iter_from_column_for_type;
4
5iter_from_column_for_type!(bool);
6iter_from_column_for_type!(i8);
7iter_from_column_for_type!(i16);
8// i32 has specific definition
9// i64 has specific definition
10iter_from_column_for_type!(u8);
11iter_from_column_for_type!(u16);
12iter_from_column_for_type!(u32);
13iter_from_column_for_type!(u64);
14iter_from_column_for_type!(f32);
15iter_from_column_for_type!(f64);
16
17#[cfg(test)]
18mod tests {
19
20    use crate::*;
21    use itertools::{izip, Itertools};
22    use polars::prelude::*;
23    use rand::{rngs::StdRng, SeedableRng};
24    use shared_test_helpers::*;
25
26    const ROW_COUNT: usize = 64;
27
28    create_test_for_chunked_type!(bool_test, bool, bool, DataType::Boolean, ROW_COUNT);
29    create_test_for_chunked_type!(i8_test, i8, i8, DataType::Int8, ROW_COUNT);
30    create_test_for_chunked_type!(i16_test, i16, i16, DataType::Int16, ROW_COUNT);
31    create_test_for_chunked_type!(i32_test, i32, i32, DataType::Int32, ROW_COUNT);
32    create_test_for_chunked_type!(u8_test, u8, u8, DataType::UInt8, ROW_COUNT);
33    create_test_for_chunked_type!(u16_test, u16, u16, DataType::UInt16, ROW_COUNT);
34    create_test_for_chunked_type!(u32_test, u32, u32, DataType::UInt32, ROW_COUNT);
35    create_test_for_chunked_type!(u64_test, u64, u64, DataType::UInt64, ROW_COUNT);
36    create_test_for_chunked_type!(f32_test, f32, f32, DataType::Float32, ROW_COUNT);
37    create_test_for_chunked_type!(f64_test, f64, f64, DataType::Float64, ROW_COUNT);
38}