vortex_array/arrays/varbinview/compute/
mask.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use vortex_error::VortexResult;
5use vortex_mask::Mask;
6
7use crate::arrays::{VarBinViewArray, VarBinViewVTable};
8use crate::compute::{MaskKernel, MaskKernelAdapter};
9use crate::vtable::ValidityHelper;
10use crate::{ArrayRef, IntoArray, register_kernel};
11
12impl MaskKernel for VarBinViewVTable {
13    fn mask(&self, array: &VarBinViewArray, mask: &Mask) -> VortexResult<ArrayRef> {
14        Ok(VarBinViewArray::try_new(
15            array.views().clone(),
16            array.buffers().clone(),
17            array.dtype().as_nullable(),
18            array.validity().mask(mask)?,
19        )?
20        .into_array())
21    }
22}
23
24register_kernel!(MaskKernelAdapter(VarBinViewVTable).lift());
25
26#[cfg(test)]
27mod tests {
28    use crate::arrays::VarBinViewArray;
29    use crate::compute::conformance::mask::test_mask_conformance;
30
31    #[test]
32    fn take_mask_var_bin_view_array() {
33        test_mask_conformance(
34            VarBinViewArray::from_iter_str(["one", "two", "three", "four", "five"]).as_ref(),
35        );
36
37        test_mask_conformance(
38            VarBinViewArray::from_iter_nullable_str([
39                Some("one"),
40                None,
41                Some("three"),
42                Some("four"),
43                Some("five"),
44            ])
45            .as_ref(),
46        );
47    }
48}