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 时,引发 class 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)