Function nullif_fuzz

arrow-select/src/nullif.rs:527–576

()

    #[test]
    fn nullif_fuzz() {
        let mut rng = StdRng::seed_from_u64(7337);

        let arrays = [
            Int32Array::from(vec![0; 1024]), // no nulls
            (0..1024) // 50% nulls
                .map(|_| rng.random_bool(0.5).then_some(1))
                .collect(),
        ];

        for a in arrays {
            let a_slices = [
                (0, 128),
                (0, 129),
                (64, 64),
                (0, 64),
                (32, 32),
                (0, 0),
                (32, 0),
                (5, 800),
                (33, 53),
                (77, 101),
            ];
            for (a_offset, a_length) in a_slices {
                let a = a.slice(a_offset, a_length);

                for i in 1..65 {
                    let b_start_offset = rng.random_range(0..i);
                    let b_end_offset = rng.random_range(0..i);

                    // b with 50% nulls
                    let b: BooleanArray = (0..a_length + b_start_offset + b_end_offset)
                        .map(|_| rng.random_bool(0.5).then(|| rng.random_bool(0.5)))
                        .collect();
                    let b_sliced = b.slice(b_start_offset, a_length);
                    test_nullif(&a, &b_sliced);

                    // b with no nulls (and no null buffer)
                    let b = remove_null_buffer(&b);
                    let b_sliced = b.slice(b_start_offset, a_length);
                    test_nullif(&a, &b_sliced);

                    // b with no nulls (but with a null buffer)
                    let b = remove_null_values(&b);
                    let b_sliced = b.slice(b_start_offset, a_length);
                    test_nullif(&a, &b_sliced);
                }
            }
        }
    }

    /// Returns a new BooleanArray with no null buffer
    fn remove_null_buffer(array: &BooleanArray) -> BooleanArray {