vortex_array/arrays/varbin/compute/
cast.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use vortex_dtype::DType;
5use vortex_error::VortexResult;
6
7use crate::arrays::{VarBinArray, VarBinVTable};
8use crate::compute::{CastKernel, CastKernelAdapter};
9use crate::vtable::ValidityHelper;
10use crate::{ArrayRef, IntoArray, register_kernel};
11
12impl CastKernel for VarBinVTable {
13    fn cast(&self, array: &VarBinArray, dtype: &DType) -> VortexResult<Option<ArrayRef>> {
14        if !array.dtype().eq_ignore_nullability(dtype) {
15            return Ok(None);
16        }
17
18        let new_nullability = dtype.nullability();
19        let new_validity = array.validity().clone().cast_nullability(new_nullability)?;
20        let new_dtype = array.dtype().with_nullability(new_nullability);
21        Ok(Some(
22            VarBinArray::try_new(
23                array.offsets().clone(),
24                array.bytes().clone(),
25                new_dtype,
26                new_validity,
27            )?
28            .into_array(),
29        ))
30    }
31}
32
33register_kernel!(CastKernelAdapter(VarBinVTable).lift());
34
35#[cfg(test)]
36mod tests {
37    use rstest::rstest;
38    use vortex_dtype::{DType, Nullability};
39
40    use crate::arrays::VarBinArray;
41    use crate::compute::cast;
42    use crate::compute::conformance::cast::test_cast_conformance;
43
44    #[rstest]
45    #[case(
46        DType::Utf8(Nullability::Nullable),
47        DType::Utf8(Nullability::NonNullable)
48    )]
49    #[case(
50        DType::Binary(Nullability::Nullable),
51        DType::Binary(Nullability::NonNullable)
52    )]
53    #[case(
54        DType::Utf8(Nullability::NonNullable),
55        DType::Utf8(Nullability::Nullable)
56    )]
57    #[case(
58        DType::Binary(Nullability::NonNullable),
59        DType::Binary(Nullability::Nullable)
60    )]
61    fn try_cast_varbin_nullable(#[case] source: DType, #[case] target: DType) {
62        let varbin = VarBinArray::from_iter(vec![Some("a"), Some("b"), Some("c")], source);
63
64        let res = cast(varbin.as_ref(), &target);
65        assert_eq!(res.unwrap().dtype(), &target);
66    }
67
68    #[rstest]
69    #[should_panic]
70    #[case(DType::Utf8(Nullability::Nullable))]
71    #[should_panic]
72    #[case(DType::Binary(Nullability::Nullable))]
73    fn try_cast_varbin_fail(#[case] source: DType) {
74        let non_nullable_source = source.as_nonnullable();
75        let varbin = VarBinArray::from_iter(vec![Some("a"), Some("b"), None], source);
76        cast(varbin.as_ref(), &non_nullable_source).unwrap();
77    }
78
79    #[rstest]
80    #[case(VarBinArray::from_iter(vec![Some("hello"), Some("world"), Some("test")], DType::Utf8(Nullability::NonNullable)))]
81    #[case(VarBinArray::from_iter(vec![Some("hello"), None, Some("world")], DType::Utf8(Nullability::Nullable)))]
82    #[case(VarBinArray::from_iter(vec![Some(b"binary".as_slice()), Some(b"data".as_slice())], DType::Binary(Nullability::NonNullable)))]
83    #[case(VarBinArray::from_iter(vec![Some(b"test".as_slice()), None], DType::Binary(Nullability::Nullable)))]
84    #[case(VarBinArray::from_iter(vec![Some("single")], DType::Utf8(Nullability::NonNullable)))]
85    fn test_cast_varbin_conformance(#[case] array: VarBinArray) {
86        test_cast_conformance(array.as_ref());
87    }
88}
vortex_array/arrays/varbin/compute/cast.rs

vortex_array/arrays/varbin/compute/
cast.rs
