Function create_random_decimal_array

arrow/src/util/data_gen.rs:204–240

(field: &Field, size: usize, null_density: f32)

#[inline]
fn create_random_decimal_array(field: &Field, size: usize, null_density: f32) -> Result<ArrayRef> {
    let mut rng = seedable_rng();

    match field.data_type() {
        DataType::Decimal128(precision, scale) => {
            let values = (0..size)
                .map(|_| {
                    if rng.random::<f32>() < null_density {
                        None
                    } else {
                        Some(rng.random::<i128>())
                    }
                })
                .collect::<Vec<_>>();
            Ok(Arc::new(
                Decimal128Array::from(values).with_precision_and_scale(*precision, *scale)?,
            ))
        }
        DataType::Decimal256(precision, scale) => {
            let values = (0..size)
                .map(|_| {
                    if rng.random::<f32>() < null_density {
                        None
                    } else {
                        Some(i256::from_parts(rng.random::<u128>(), rng.random::<i128>()))
                    }
                })
                .collect::<Vec<_>>();
            Ok(Arc::new(
                Decimal256Array::from(values).with_precision_and_scale(*precision, *scale)?,
            ))
        }
        _ => Err(ArrowError::InvalidArgumentError(format!(
            "Cannot create decimal array for field {field}"
        ))),
    }
}
#[inline]
fn create_random_run_end_encoded_array(
    index: &Field,