numpy.testing.assert_warns

testing.assert_warns(warning_class, *args, **kwargs) [source]

Проверка, что вызываемая функция генерирует указанное предупреждение.

При вызове вызываемой функции с аргументами args и ключевыми аргументами kwargs должно быть сгенерировано предупреждение класса warning_class. Если будет сгенерировано предупреждение другого типа, оно не будет поймано.

Если вызывается со всеми аргументами, кроме класса предупреждения, можно использовать как менеджер контекста:

с assert_warns(SomeWarning):

do_something()

Возможность использования в качестве менеджера контекста появилась в NumPy v1.11.0.

Новое в версии 1.4.0.

Параметры

warning_classclass

Класс, определяющий предупреждение, которое func ожидается выбросить.

funccallable, optional

Вызываемая функция для проверки

*argsArguments

Аргументы для func.

**kwargsKwargs

Ключевые аргументы для func.

Возвращает

The value returned by 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