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::ArrayRef;
8use crate::IntoArray;
9use crate::arrays::VarBinArray;
10use crate::arrays::VarBinVTable;
11use crate::compute::CastReduce;
12use crate::vtable::ValidityHelper;
13
14impl CastReduce for VarBinVTable {
15    fn cast(array: &VarBinArray, 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        Ok(Some(
27            VarBinArray::try_new(
28                array.offsets().clone(),
29                array.bytes().clone(),
30                new_dtype,
31                new_validity,
32            )?
33            .into_array(),
34        ))
35    }
36}
37
38#[cfg(test)]
39mod tests {
40    use rstest::rstest;
41    use vortex_dtype::DType;
42    use vortex_dtype::Nullability;
43
44    use crate::arrays::VarBinArray;
45    use crate::builtins::ArrayBuiltins;
46    use crate::compute::conformance::cast::test_cast_conformance;
47
48    #[rstest]
49    #[case(
50        DType::Utf8(Nullability::Nullable),
51        DType::Utf8(Nullability::NonNullable)
52    )]
53    #[case(
54        DType::Binary(Nullability::Nullable),
55        DType::Binary(Nullability::NonNullable)
56    )]
57    #[case(
58        DType::Utf8(Nullability::NonNullable),
59        DType::Utf8(Nullability::Nullable)
60    )]
61    #[case(
62        DType::Binary(Nullability::NonNullable),
63        DType::Binary(Nullability::Nullable)
64    )]
65    fn try_cast_varbin_nullable(#[case] source: DType, #[case] target: DType) {
66        let varbin = VarBinArray::from_iter(vec![Some("a"), Some("b"), Some("c")], source);
67
68        let res = varbin.to_array().cast(target.clone());
69        assert_eq!(res.unwrap().dtype(), &target);
70    }
71
72    #[rstest]
73    #[should_panic]
74    #[case(DType::Utf8(Nullability::Nullable))]
75    #[should_panic]
76    #[case(DType::Binary(Nullability::Nullable))]
77    fn try_cast_varbin_fail(#[case] source: DType) {
78        let non_nullable_source = source.as_nonnullable();
79        let varbin = VarBinArray::from_iter(vec![Some("a"), Some("b"), None], source);
80        varbin.to_array().cast(non_nullable_source).unwrap();
81    }
82
83    #[rstest]
84    #[case(VarBinArray::from_iter(vec![Some("hello"), Some("world"), Some("test")], DType::Utf8(Nullability::NonNullable)))]
85    #[case(VarBinArray::from_iter(vec![Some("hello"), None, Some("world")], DType::Utf8(Nullability::Nullable)))]
86    #[case(VarBinArray::from_iter(vec![Some(b"binary".as_slice()), Some(b"data".as_slice())], DType::Binary(Nullability::NonNullable)))]
87    #[case(VarBinArray::from_iter(vec![Some(b"test".as_slice()), None], DType::Binary(Nullability::Nullable)))]
88    #[case(VarBinArray::from_iter(vec![Some("single")], DType::Utf8(Nullability::NonNullable)))]
89    fn test_cast_varbin_conformance(#[case] array: VarBinArray) {
90        test_cast_conformance(array.as_ref());
91    }
92}
vortex_array/arrays/varbin/compute/cast.rs

vortex_array/arrays/varbin/compute/
cast.rs
