vortex_array/arrays/bool/vtable/
operations.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use vortex_error::VortexResult;
5
6use crate::ExecutionCtx;
7use crate::array::ArrayView;
8use crate::array::OperationsVTable;
9use crate::arrays::Bool;
10use crate::arrays::bool::BoolArrayExt;
11use crate::scalar::Scalar;
12
13impl OperationsVTable<Bool> for Bool {
14    fn scalar_at(
15        array: ArrayView<'_, Bool>,
16        index: usize,
17        _ctx: &mut ExecutionCtx,
18    ) -> VortexResult<Scalar> {
19        Ok(Scalar::bool(
20            array.bit_buffer_view().value(index),
21            array.dtype().nullability(),
22        ))
23    }
24}
25
26#[cfg(test)]
27mod tests {
28    use std::iter;
29
30    use crate::IntoArray;
31    #[expect(deprecated)]
32    use crate::ToCanonical as _;
33    use crate::VortexSessionExecute;
34    use crate::array_session;
35    use crate::arrays::BoolArray;
36    use crate::arrays::bool::BoolArrayExt;
37    use crate::assert_arrays_eq;
38
39    #[test]
40    fn test_slice_hundred_elements() {
41        let arr = BoolArray::from_iter(iter::repeat_n(Some(true), 100));
42        #[expect(deprecated)]
43        let sliced_arr = arr.into_array().slice(8..16).unwrap().to_bool();
44        assert_eq!(sliced_arr.len(), 8);
45        assert_eq!(sliced_arr.to_bit_buffer().len(), 8);
46        assert_eq!(sliced_arr.to_bit_buffer().offset(), 0);
47    }
48
49    #[test]
50    fn test_slice() {
51        let mut ctx = array_session().create_execution_ctx();
52        let arr = BoolArray::from_iter([Some(true), Some(true), None, Some(false), None]);
53        #[expect(deprecated)]
54        let sliced_arr = arr.into_array().slice(1..4).unwrap().to_bool();
55
56        assert_arrays_eq!(
57            sliced_arr,
58            BoolArray::from_iter([Some(true), None, Some(false)]),
59            &mut ctx
60        );
61    }
62}
vortex_array/arrays/bool/vtable/operations.rs

vortex_array/arrays/bool/vtable/
operations.rs
