Function polars::export::arrow::compute::arithmetics::decimal::checked_add
source · [−]pub fn checked_add(
lhs: &PrimitiveArray<i128>,
rhs: &PrimitiveArray<i128>
) -> PrimitiveArray<i128>
This is supported on crate feature
compute_arithmetics
only.Expand description
Checked addition of two decimal primitive arrays with the same precision and scale. If the precision and scale is different, then an InvalidArgumentError is returned. If the result from the sum is larger than the possible number with the selected precision (overflowing), then the validity for that index is changed to None
Examples
use arrow2::compute::arithmetics::decimal::checked_add;
use arrow2::array::PrimitiveArray;
use arrow2::datatypes::DataType;
let a = PrimitiveArray::from([Some(99000i128), Some(11100i128), None, Some(22200i128)]).to(DataType::Decimal(5, 2));
let b = PrimitiveArray::from([Some(01000i128), Some(22200i128), None, Some(11100i128)]).to(DataType::Decimal(5, 2));
let result = checked_add(&a, &b);
let expected = PrimitiveArray::from([None, Some(33300i128), None, Some(33300i128)]).to(DataType::Decimal(5, 2));
assert_eq!(result, expected);