Python numpy testing.assert_warns用法及代码示例

本文简要介绍 python 语言中 numpy.testing.assert_warns 的用法。

用法:

testing.assert_warns(warning_class, *args, **kwargs)

除非给定的可调用对象抛出指定的警告，否则失败。

当使用参数 args 和关键字参数 kwargs 调用时，可调用对象应抛出类 warning_class 的警告。如果抛出不同类型的警告，则不会被捕获。

如果调用时省略了警告类以外的所有参数，则可以用作上下文管理器：

with assert_warns(SomeWarning):

do_something()

NumPy v1.11.0 中新增了用作上下文管理器的函数。

参数：

warning_class： 类

定义 func 应该抛出的警告的类。

func： 可调用的，可选的

可调用测试

*args： 参数

函数的参数。

**kwargs： 夸格斯

func 的关键字参数。

返回：

返回的值函数.

例子：

>>> import warnings
>>> def deprecated_func(num):
...     warnings.warn("Please upgrade", DeprecationWarning)
...     return num*num
>>> with np.testing.assert_warns(DeprecationWarning):
...     assert deprecated_func(4) == 16
>>> # or passing a func
>>> ret = np.testing.assert_warns(DeprecationWarning, deprecated_func, 4)
>>> assert ret == 16