Function arb_array

src/repr/src/scalar.rs:5516–5543

(element_strategy: BoxedStrategy<PropDatum>)

#[cfg(any(test, feature = "proptest"))]
fn arb_array(element_strategy: BoxedStrategy<PropDatum>) -> BoxedStrategy<PropArray> {
    // Elements in Arrays can always be Null.
    let element_strategy = Union::new_weighted(vec![
        (20, element_strategy),
        (1, Just(PropDatum::Null).boxed()),
    ]);

    prop::collection::vec(
        arb_array_dimension(),
        1..usize::from(crate::adt::array::MAX_ARRAY_DIMENSIONS),
    )
    .prop_flat_map(move |dimensions| {
        let n_elts: usize = dimensions.iter().map(|d| d.length).product();
        (
            Just(dimensions),
            prop::collection::vec(element_strategy.clone(), n_elts),
        )
    })
    .prop_map(|(dimensions, elements)| {
        let element_datums: Vec<Datum<'_>> = elements.iter().map(|pd| pd.into()).collect();
        let mut row = Row::default();
        row.packer()
            .try_push_array(&dimensions, element_datums)
            .unwrap();
        PropArray(row, elements)
    })
    .boxed()
}

#[derive(Debug, PartialEq, Clone)]
#[cfg(any(test, feature = "proptest"))]