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.to_bit_buffer().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::arrays::BoolArray;
34    use crate::arrays::bool::BoolArrayExt;
35    use crate::assert_arrays_eq;
36
37    #[test]
38    fn test_slice_hundred_elements() {
39        let arr = BoolArray::from_iter(iter::repeat_n(Some(true), 100));
40        #[expect(deprecated)]
41        let sliced_arr = arr.into_array().slice(8..16).unwrap().to_bool();
42        assert_eq!(sliced_arr.len(), 8);
43        assert_eq!(sliced_arr.to_bit_buffer().len(), 8);
44        assert_eq!(sliced_arr.to_bit_buffer().offset(), 0);
45    }
46
47    #[test]
48    fn test_slice() {
49        let arr = BoolArray::from_iter([Some(true), Some(true), None, Some(false), None]);
50        #[expect(deprecated)]
51        let sliced_arr = arr.into_array().slice(1..4).unwrap().to_bool();
52
53        assert_arrays_eq!(
54            sliced_arr,
55            BoolArray::from_iter([Some(true), None, Some(false)])
56        );
57    }
58}
vortex_array/arrays/bool/vtable/operations.rs

vortex_array/arrays/bool/vtable/
operations.rs
