Function test_sliced_list_offsets

datafusion/functions-nested/src/array_has.rs:1237–1276

()

    #[test]
    fn test_sliced_list_offsets() {
        // Full rows:
        //   row 0: [1, 2]   (not visible after slicing)
        //   row 1: [11, 12] (visible row 0)
        //   row 2: [21, 22] (visible row 1)
        //   row 3: [31, 32] (not visible after slicing)
        let field: Arc<Field> = Arc::new(Field::new("item", DataType::Int32, false));
        let full_values = Arc::new(Int32Array::from(vec![1, 2, 11, 12, 21, 22, 31, 32]));
        let full_offsets = OffsetBuffer::new(vec![0, 2, 4, 6, 8].into());
        let full = ListArray::new(Arc::clone(&field), full_offsets, full_values, None);
        let sliced_haystack: ArrayRef = Arc::new(full.slice(1, 2));

        // array_has_all: needle row 0 = [11], row 1 = [21]
        let needle_all: ArrayRef = Arc::new(ListArray::new(
            Arc::clone(&field),
            OffsetBuffer::new(vec![0, 1, 2].into()),
            Arc::new(Int32Array::from(vec![11, 21])),
            None,
        ));
        invoke_and_assert(
            &ArrayHasAll::new(),
            &sliced_haystack,
            needle_all,
            &[Some(true), Some(true)],
        );

        // array_has_any: needle row 0 = [99, 11], row 1 = [99, 21]
        let needle_any: ArrayRef = Arc::new(ListArray::new(
            field,
            OffsetBuffer::new(vec![0, 2, 4].into()),
            Arc::new(Int32Array::from(vec![99, 11, 99, 21])),
            None,
        ));
        invoke_and_assert(
            &ArrayHasAny::new(),
            &sliced_haystack,
            needle_any,
            &[Some(true), Some(true)],
        );
    }

    #[test]
    fn test_sliced_fixed_size_list_offsets() {