vortex_array/arrays/constant/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::ConstantArray;
10use crate::arrays::ConstantVTable;
11use crate::compute::CastKernel;
12use crate::compute::CastKernelAdapter;
13use crate::register_kernel;
14
15impl CastKernel for ConstantVTable {
16    fn cast(&self, array: &ConstantArray, dtype: &DType) -> VortexResult<Option<ArrayRef>> {
17        match array.scalar().cast(dtype) {
18            Ok(scalar) => Ok(Some(ConstantArray::new(scalar, array.len()).into_array())),
19            Err(_e) => Ok(None),
20        }
21    }
22}
23
24register_kernel!(CastKernelAdapter(ConstantVTable).lift());
25
26#[cfg(test)]
27mod tests {
28    use rstest::rstest;
29    use vortex_scalar::Scalar;
30
31    use crate::IntoArray;
32    use crate::arrays::ConstantArray;
33    use crate::compute::conformance::cast::test_cast_conformance;
34
35    #[rstest]
36    #[case(ConstantArray::new(Scalar::from(42u32), 5).into_array())]
37    #[case(ConstantArray::new(Scalar::from(-100i32), 10).into_array())]
38    #[case(ConstantArray::new(Scalar::from(3.5f32), 3).into_array())]
39    #[case(ConstantArray::new(Scalar::from(true), 7).into_array())]
40    #[case(ConstantArray::new(Scalar::null_typed::<i32>(), 4).into_array())]
41    #[case(ConstantArray::new(Scalar::from(255u8), 1).into_array())]
42    fn test_cast_constant_conformance(#[case] array: crate::ArrayRef) {
43        test_cast_conformance(array.as_ref());
44    }
45}
vortex_array/arrays/constant/compute/cast.rs

vortex_array/arrays/constant/compute/
cast.rs
