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