numpy.testing.assert_raises

testing.assert_raises(exception_class, callable, *args, **kwargs) assert_raises(exception_class)

testing.assert_raises(exception_class) → None

除非使用参数 args 和关键字参数 kwargs 调用 callable 时抛出 exception_class 类的异常,否则失败.如果抛出其他类型的异常,它将不会被捕获,并且测试用例将被视为遭受了错误,就像出现意外异常一样.

或者, assert_raises 可以用作上下文管理器:

>>> from numpy.testing import assert_raises
>>> with assert_raises(ZeroDivisionError):
...     1 / 0

等价于

>>> def div(x, y):
...     return x / y
>>> assert_raises(ZeroDivisionError, div, 1, 0)