vortex_array/arrays/varbinview/compute/
cast.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use vortex_error::VortexResult;
5
6use crate::ArrayRef;
7use crate::IntoArray;
8use crate::arrays::VarBinViewArray;
9use crate::arrays::VarBinViewVTable;
10use crate::dtype::DType;
11use crate::scalar_fn::fns::cast::CastReduce;
12use crate::vtable::ValidityHelper;
13
14impl CastReduce for VarBinViewVTable {
15    fn cast(array: &VarBinViewArray, dtype: &DType) -> VortexResult<Option<ArrayRef>> {
16        if !array.dtype().eq_ignore_nullability(dtype) {
17            return Ok(None);
18        }
19
20        let new_nullability = dtype.nullability();
21        let new_validity = array
22            .validity()
23            .clone()
24            .cast_nullability(new_nullability, array.len())?;
25        let new_dtype = array.dtype().with_nullability(new_nullability);
26
27        // SAFETY: casting just changes the DType, does not affect invariants on views/buffers.
28        unsafe {
29            Ok(Some(
30                VarBinViewArray::new_handle_unchecked(
31                    array.views_handle().clone(),
32                    array.buffers().clone(),
33                    new_dtype,
34                    new_validity,
35                )
36                .into_array(),
37            ))
38        }
39    }
40}
41
42#[cfg(test)]
43mod tests {
44    use rstest::rstest;
45
46    use crate::IntoArray;
47    use crate::arrays::VarBinViewArray;
48    use crate::builtins::ArrayBuiltins;
49    use crate::compute::conformance::cast::test_cast_conformance;
50    use crate::dtype::DType;
51    use crate::dtype::Nullability;
52
53    #[rstest]
54    #[case(
55        DType::Utf8(Nullability::Nullable),
56        DType::Utf8(Nullability::NonNullable)
57    )]
58    #[case(
59        DType::Binary(Nullability::Nullable),
60        DType::Binary(Nullability::NonNullable)
61    )]
62    #[case(
63        DType::Utf8(Nullability::NonNullable),
64        DType::Utf8(Nullability::Nullable)
65    )]
66    #[case(
67        DType::Binary(Nullability::NonNullable),
68        DType::Binary(Nullability::Nullable)
69    )]
70    fn try_cast_varbin_nullable(#[case] source: DType, #[case] target: DType) {
71        let varbin = VarBinViewArray::from_iter(vec![Some("a"), Some("b"), Some("c")], source);
72
73        let res = varbin.into_array().cast(target.clone());
74        assert_eq!(res.unwrap().dtype(), &target);
75    }
76
77    #[rstest]
78    #[should_panic]
79    #[case(DType::Utf8(Nullability::Nullable))]
80    #[should_panic]
81    #[case(DType::Binary(Nullability::Nullable))]
82    fn try_cast_varbin_fail(#[case] source: DType) {
83        let non_nullable_source = source.as_nonnullable();
84        let varbin = VarBinViewArray::from_iter(vec![Some("a"), Some("b"), None], source);
85        varbin.into_array().cast(non_nullable_source).unwrap();
86    }
87
88    #[rstest]
89    #[case(VarBinViewArray::from_iter(vec![Some("hello"), Some("world"), Some("test")], DType::Utf8(Nullability::NonNullable)))]
90    #[case(VarBinViewArray::from_iter(vec![Some("hello"), None, Some("world")], DType::Utf8(Nullability::Nullable)))]
91    #[case(VarBinViewArray::from_iter(vec![Some(b"binary".as_slice()), Some(b"data".as_slice())], DType::Binary(Nullability::NonNullable)))]
92    #[case(VarBinViewArray::from_iter(vec![Some(b"test".as_slice()), None], DType::Binary(Nullability::Nullable)))]
93    #[case(VarBinViewArray::from_iter(vec![Some("single")], DType::Utf8(Nullability::NonNullable)))]
94    #[case(VarBinViewArray::from_iter(vec![Some("very long string that exceeds the inline size to test view functionality with multiple buffers")], DType::Utf8(Nullability::NonNullable)))]
95    fn test_cast_varbinview_conformance(#[case] array: VarBinViewArray) {
96        test_cast_conformance(&array.into_array());
97    }
98}
vortex_array/arrays/varbinview/compute/cast.rs

vortex_array/arrays/varbinview/compute/
cast.rs
