Function test_error_len_mismatch

pandas/tests/arrays/masked/test_arithmetic.py:192–225

(data, all_arithmetic_operators)

def test_error_len_mismatch(data, all_arithmetic_operators):
    # operating with a list-like with non-matching length raises
    data, scalar = data
    op = tm.get_op_from_name(all_arithmetic_operators)

    other = [scalar] * (len(data) - 1)

    err = ValueError
    msg = "|".join(
        [
            r"operands could not be broadcast together with shapes \(3,\) \(4,\)",
            r"operands could not be broadcast together with shapes \(4,\) \(3,\)",
        ]
    )
    if data.dtype.kind == "b" and all_arithmetic_operators.strip("_") in [
        "sub",
        "rsub",
    ]:
        err = TypeError
        msg = (
            r"numpy boolean subtract, the `\-` operator, is not supported, use "
            r"the bitwise_xor, the `\^` operator, or the logical_xor function instead"
        )
    elif is_bool_not_implemented(data, all_arithmetic_operators):
        msg = "operator '.*' not implemented for bool dtypes"
        err = NotImplementedError

    for val in [other, np.array(other)]:
        with pytest.raises(err, match=msg):
            op(data, val)

        s = pd.Series(data)
        with pytest.raises(err, match=msg):
            op(s, val)


@pytest.mark.parametrize("op", ["__neg__", "__abs__", "__invert__"])