Function assert_warns

dali/test/python/nose_utils.py:136–168

Wrapper for asserting warnings, optionally with pattern matching. Can be used as context manager or with callable: with assert_warns(UserWarning): warnings.warn("test", UserWarning) assert_warns(UserWarning, callable, arg1, arg2, kwarg=value)

(exception=Warning, *args, glob=None, regex=None, match_case=None, **kwargs)

def assert_warns(exception=Warning, *args, glob=None, regex=None, match_case=None, **kwargs):
    """
    Wrapper for asserting warnings, optionally with pattern matching.

    Can be used as context manager or with callable:
        with assert_warns(UserWarning):
            warnings.warn("test", UserWarning)

        assert_warns(UserWarning, callable, arg1, arg2, kwarg=value)
    """
    if glob is None and regex is None:
        # Use unittest's assertWarns
        if args:
            # Called with callable
            callable_func = args[0]
            callable_args = args[1:]
            with _test_case.assertWarns(exception):
                callable_func(*callable_args, **kwargs)
        else:
            # Used as context manager
            return _test_case.assertWarns(exception)
    else:
        pattern = get_pattern(glob, regex, match_case)
        # Use unittest's assertWarnsRegex
        if args:
            # Called with callable
            callable_func = args[0]
            callable_args = args[1:]
            with _test_case.assertWarnsRegex(exception, pattern):
                callable_func(*callable_args, **kwargs)
        else:
            # Used as context manager
            return _test_case.assertWarnsRegex(exception, pattern)


def raises(exception, glob=None, regex=None, match_case=None):