def METHOD_NAME():
    # Models as inputs
    result = _arith_oper(sh1, scl1)
    assert_allclose(result, np.array([[1]]))
    result = _arith_oper(rot, rot)
    assert_allclose(result, np.array([[1, 1], [1, 1]]))
    # ndarray
    result = _arith_oper(np.array([[1, 2], [3, 4]]), np.array([[1, 2], [3, 4]]))
    assert_allclose(result, np.array([[1, 1], [1, 1]]))
    # Error
    with pytest.raises(
        ModelDefinitionError, match=r"Unsupported operands for arithmetic operator: .*"
    ):
        _arith_oper(sh1, map1)