jupytersketcher/Lib/site-packages/joblib/test/test_my_exceptions.py

"""
Test my automatically generate exceptions
"""
from joblib.my_exceptions import (
    JoblibException, JoblibNameError, _mk_exception)


class CustomException(Exception):
    def __init__(self, a, b, c, d):
        self.a, self.b, self.c, self.d = a, b, c, d


class CustomException2(Exception):
    """A custom exception with a .args attribute

    Just to check that the JoblibException created from it
    has it args set correctly
    """
    def __init__(self, a, *args):
        self.a = a
        self.args = args


def test_inheritance():
    assert isinstance(JoblibNameError(), NameError)
    assert isinstance(JoblibNameError(), JoblibException)
    assert (JoblibNameError is _mk_exception(NameError)[0])


def test_inheritance_special_cases():
    # _mk_exception should transform Exception to JoblibException
    assert (_mk_exception(Exception)[0] is JoblibException)

    # Subclasses of JoblibException should be mapped to
    # them-selves by _mk_exception
    assert (_mk_exception(JoblibException)[0] is JoblibException)

    # Non-inheritable exception classes should be mapped to
    # JoblibException by _mk_exception. That can happen with classes
    # generated with SWIG. See
    # https://github.com/joblib/joblib/issues/269 for a concrete
    # example.
    non_inheritable_classes = [type(lambda: None), bool]
    for exception in non_inheritable_classes:
        assert (_mk_exception(exception)[0] is JoblibException)


def test__mk_exception():
    # Check that _mk_exception works on a bunch of different exceptions
    for klass in (Exception, TypeError, SyntaxError, ValueError,
                  ImportError, CustomException, CustomException2):
        message = 'This message should be in the exception repr'
        exc = _mk_exception(klass)[0](
            message, 'some', 'other', 'args', 'that are not', 'in the repr')
        exc_repr = repr(exc)

        assert isinstance(exc, klass)
        assert isinstance(exc, JoblibException)
        assert exc.__class__.__name__ in exc_repr
        assert message in exc_repr