vortex-array 0.62.0

Vortex in memory columnar data format
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
55
56
57
58
59
60
61
62
63
64
65
66

// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: Copyright the Vortex contributors

mod between;
mod cast;
mod fill_null;
mod filter;
mod min_max;
mod not;
pub(crate) mod rules;
mod slice;
mod sum;
mod take;

#[cfg(test)]
mod test {
    use rstest::rstest;

    use crate::IntoArray;
    use crate::arrays::ConstantArray;
    use crate::compute::conformance::consistency::test_array_consistency;
    use crate::compute::conformance::filter::test_filter_conformance;
    use crate::compute::conformance::mask::test_mask_conformance;
    use crate::dtype::half::f16;
    use crate::scalar::Scalar;

    #[test]
    fn test_mask_constant() {
        test_mask_conformance(&ConstantArray::new(Scalar::null_native::<i32>(), 5).into_array());
        test_mask_conformance(&ConstantArray::new(Scalar::from(3u16), 5).into_array());
        test_mask_conformance(&ConstantArray::new(Scalar::from(1.0f32 / 0.0f32), 5).into_array());
        test_mask_conformance(
            &ConstantArray::new(Scalar::from(f16::from_f32(3.0f32)), 5).into_array(),
        );
    }

    #[test]
    fn test_filter_constant() {
        test_filter_conformance(&ConstantArray::new(Scalar::null_native::<i32>(), 5).into_array());
        test_filter_conformance(&ConstantArray::new(Scalar::from(3u16), 5).into_array());
        test_filter_conformance(&ConstantArray::new(Scalar::from(1.0f32 / 0.0f32), 5).into_array());
        test_filter_conformance(
            &ConstantArray::new(Scalar::from(f16::from_f32(3.0f32)), 5).into_array(),
        );
    }

    #[rstest]
    // From test_all_consistency
    #[case::constant_i32(ConstantArray::new(Scalar::from(42i32), 5))]
    #[case::constant_str(ConstantArray::new(Scalar::from("constant"), 5))]
    #[case::constant_null(ConstantArray::new(
        Scalar::null(crate::dtype::DType::Primitive(
            crate::dtype::PType::I32,
            crate::dtype::Nullability::Nullable
        )),
        5
    ))]
    // Additional test cases
    #[case::constant_f64(ConstantArray::new(Scalar::from(std::f64::consts::PI), 10))]
    #[case::constant_bool(ConstantArray::new(Scalar::from(true), 7))]
    #[case::constant_single(ConstantArray::new(Scalar::from(99u64), 1))]
    #[case::constant_large(ConstantArray::new(Scalar::from("hello"), 1000))]
    fn test_constant_consistency(#[case] array: ConstantArray) {
        test_array_consistency(&array.into_array());
    }
}