vortex-pco 0.65.0

Vortex Pco array
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
67

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

mod cast;

#[cfg(test)]
mod tests {
    use rstest::rstest;
    use vortex_array::IntoArray;
    use vortex_array::arrays::PrimitiveArray;
    use vortex_array::compute::conformance::consistency::test_array_consistency;

    use crate::PcoArray;

    fn pco_f32() -> PcoArray {
        let values = PrimitiveArray::from_iter([1.23f32, 4.56, 7.89, 10.11, 12.13]);
        PcoArray::from_primitive(&values, 0, 128).unwrap()
    }

    fn pco_f64() -> PcoArray {
        let values = PrimitiveArray::from_iter([100.1f64, 200.2, 300.3, 400.4, 500.5]);
        PcoArray::from_primitive(&values, 0, 128).unwrap()
    }

    fn pco_i32() -> PcoArray {
        let values = PrimitiveArray::from_iter([100i32, 200, 300, 400, 500]);
        PcoArray::from_primitive(&values, 0, 128).unwrap()
    }

    fn pco_u64() -> PcoArray {
        let values = PrimitiveArray::from_iter([1000u64, 2000, 3000, 4000]);
        PcoArray::from_primitive(&values, 0, 128).unwrap()
    }

    fn pco_i16() -> PcoArray {
        let values = PrimitiveArray::from_iter([10i16, 20, 30, 40, 50]);
        PcoArray::from_primitive(&values, 0, 128).unwrap()
    }

    fn pco_i32_alt() -> PcoArray {
        let values = PrimitiveArray::from_iter([1i32, 2, 3, 4, 5]);
        PcoArray::from_primitive(&values, 0, 128).unwrap()
    }

    fn pco_single() -> PcoArray {
        let values = PrimitiveArray::from_iter([42.42f64]);
        PcoArray::from_primitive(&values, 0, 128).unwrap()
    }

    fn pco_large() -> PcoArray {
        let values = PrimitiveArray::from_iter(0u32..1000);
        PcoArray::from_primitive(&values, 3, 128).unwrap()
    }

    #[rstest]
    #[case::f32(pco_f32())]
    #[case::f64(pco_f64())]
    #[case::i32(pco_i32())]
    #[case::u64(pco_u64())]
    #[case::i16(pco_i16())]
    #[case::i32_alt(pco_i32_alt())]
    #[case::single(pco_single())]
    #[case::large(pco_large())]
    fn test_pco_consistency(#[case] array: PcoArray) {
        test_array_consistency(&array.into_array());
    }
}