让Python的`assert`抛出我选择的异常

Question

我可以让 assert 抛出我选择的异常而不是 AssertionError 吗？

更新：

我将解释我的动机：到目前为止，我已经进行了断言式测试，这些测试引发了我自己的异常；例如，当您使用某些参数创建 Node 对象时，它会检查这些参数是否适合创建节点，如果不适合，则会引发 NodeError。

但我知道 Python 有一个 -o 模式，在该模式下断言会被跳过，我希望能够使用它，因为它会让我的程序更快。但我仍然希望有自己的例外。这就是为什么我想在我自己的异常中使用断言。

Answer 1

这会起作用。但这有点疯狂。

try:
    assert False, "A Message"
except AssertionError, e:
    raise Exception( e.args )

为什么不是下面的呢？这不那么疯狂。

if not someAssertion: raise Exception( "Some Message" )

它只比 assert 语句稍微冗长一些，但并不违反我们的预期，即断言失败会引发 AssertionError。

考虑一下这一点。

def myAssert( condition, action ):
    if not condition: raise action

然后你或多或少可以用这样的东西替换你现有的断言。

myAssert( {{ the original condition }}, MyException( {{ the original message }} ) )

完成此操作后，您现在可以自由地启用或禁用或任何您想要执行的操作。

另外，请阅读 警告 模块。这可能正是您想要做的。

Answer 2

这个怎么样？

>>> def myraise(e): raise e
... 
>>> cond=False
>>> assert cond or myraise(RuntimeError)
Traceback (most recent call last):
  File "", line 1, in 
  File "", line 1, in myraise
RuntimeError

Answer 3

切勿将断言用于逻辑！仅用于可选测试检查。请记住，如果 Python 在打开优化的情况下运行，则断言甚至不会编译到字节码中。如果你这样做，你显然关心引发的异常，如果你关心，那么你首先就使用了错误的断言。

Answer 4

当使用 -o 选项运行时，Python 也会跳过 if __debug__: 块。下面的代码比较冗长，但无需黑客

def my_assert(condition, message=None):
    if not condition:
        raise MyAssertError(message)

if __debug__: my_assert(condition, message)

即可完成您需要的操作：您可以通过在 my_assert() 内移动 if __debug__: 条件来缩短代码，但随后它会启用优化时调用（内部没有任何操作）。

Answer 5

您可以让 上下文管理器 在 with 中为您进行转换块（可能包含多个断言，或者更多代码和函数调用或您想要的内容。

from __future__ import with_statement
import contextlib

@contextlib.contextmanager
def myassert(exctype):
    try:
        yield
    except AssertionError, exc:
        raise exctype(*exc.args)

with myassert(ValueError):
    assert 0, "Zero is bad for you"

请参阅此答案的先前版本以直接替换构造的异常对象（KeyError("bad key")） ，而不是重用断言的参数。

Answer 6

至少在 Python 2.6.3 中，这也可以工作：

class MyAssertionError (Exception):
    pass

AssertionError = MyAssertionError

assert False, "False"

---

Traceback (most recent call last):
  File "assert.py", line 8, in <module>
    assert False, "False"
__main__.MyAssertionError: False

Answer 7

如果你想为此使用断言，这似乎工作得很好：

>>> def raise_(e): raise e
...
>>> x = -2
>>> assert x >= 0, raise_(ValueError('oops'))
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in raise_
ValueError: oops

请注意，在断言中，只有在条件为 false 时才会评估逗号后面的内容，因此 ValueError 只会是在需要时创建并提出。

Answer 8

为了看看尝试是否有任何开销，我

在这里尝试了这个实验是 myassert.py

def myassert(e):
    raise e


def f1(): #这是实验的对照
    cond=True

 def f2():
    条件=真
    尝试：
        断言条件，“消息”
    除了断言错误，e：
        引发异常（e.args）

 def f3（）：
    条件=真
    断言 cond 或 myassert(RuntimeError)

 def f4():
    条件=真
    如果__调试__：
        引发（运行时错误）

$ python -O -mtimeit -n100 -r1000 -s'import myassert' 'myassert.f1()'
100 loops, best of 1000: 0.42 usec per loop
$ python -O -mtimeit -n100 -r1000 -s'import myassert' 'myassert.f2()'
100 loops, best of 1000: 0.479 usec per loop
$ python -O -mtimeit -n100 -r1000 -s'import myassert' 'myassert.f3()'
100 loops, best of 1000: 0.42 usec per loop
$ python -O -mtimeit -n100 -r1000 -s'import myassert' 'myassert.f4()'
100 loops, best of 1000: 0.42 usec per loop