60 lines
2 KiB
Python
60 lines
2 KiB
Python
"""
|
|
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
|