Function assert_raises

dali/test/python/nose_utils.py:95–133

Wrapper combining unittest assertRaises and assertRaisesRegex. Specify ``regex=pattern`` or ``glob=pattern`` to check error message of expected exception against the pattern. Value for `glob` must be a string, `regex` can be either a literal or compiled regex pattern. By default

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

def assert_raises(exception, *args, glob=None, regex=None, match_case=None, **kwargs):
    """
    Wrapper combining unittest assertRaises and assertRaisesRegex.
    Specify ``regex=pattern`` or ``glob=pattern`` to check error message of expected exception
    against the pattern.
    Value for `glob` must be a string, `regex` can be either a literal or compiled regex pattern.
    By default, the check will ignore case, if called with `glob` or a literal for `regex`.
    To enforce case sensitive check pass ``match_case=True``.
    Don't specify `match_case` if passing already compiled regex pattern.

    Can be used as context manager or with callable:
        with assert_raises(Exception):
            raise Exception()

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


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