Function test_eq_array_run_end_encoded

datafusion/common/src/scalar/mod.rs:10492–10532

()

    #[test]
    fn test_eq_array_run_end_encoded() {
        let run_ends = Int16Array::from(vec![1, 3]);
        let values = Float32Array::from(vec![None, Some(1.0)]);
        let run_array =
            Arc::new(RunArray::try_new(&run_ends, &values).unwrap()) as ArrayRef;

        let scalar = ScalarValue::RunEndEncoded(
            Field::new("run_ends", DataType::Int16, false).into(),
            Field::new("values", DataType::Float32, true).into(),
            Box::new(ScalarValue::Float32(None)),
        );
        assert!(scalar.eq_array(&run_array, 0).unwrap());

        let scalar = ScalarValue::RunEndEncoded(
            Field::new("run_ends", DataType::Int16, false).into(),
            Field::new("values", DataType::Float32, true).into(),
            Box::new(ScalarValue::Float32(Some(1.0))),
        );
        assert!(scalar.eq_array(&run_array, 1).unwrap());
        assert!(scalar.eq_array(&run_array, 2).unwrap());

        // value types must match
        let scalar = ScalarValue::RunEndEncoded(
            Field::new("run_ends", DataType::Int16, false).into(),
            Field::new("values", DataType::Float64, true).into(),
            Box::new(ScalarValue::Float64(Some(1.0))),
        );
        let err = scalar.eq_array(&run_array, 1).unwrap_err();
        let expected = "Internal error: could not cast array of type Float32 to arrow_array::array::primitive_array::PrimitiveArray<arrow_array::types::Float64Type>";
        assert!(err.to_string().starts_with(expected));

        // run ends type must match
        let scalar = ScalarValue::RunEndEncoded(
            Field::new("run_ends", DataType::Int32, false).into(),
            Field::new("values", DataType::Float32, true).into(),
            Box::new(ScalarValue::Float32(None)),
        );
        let err = scalar.eq_array(&run_array, 0).unwrap_err();
        let expected = "Internal error: could not cast array of type RunEndEncoded(\"run_ends\": non-null Int16, \"values\": Float32) to arrow_array::array::run_array::RunArray<arrow_array::types::Int32Type>";
        assert!(err.to_string().starts_with(expected));
    }

    #[test]
    fn test_iter_to_array_run_end_encoded() {