diff options
-rw-r--r-- | ipalib/errors.py | 2 | ||||
-rw-r--r-- | ipalib/tests/test_errors.py | 10 |
2 files changed, 7 insertions, 5 deletions
diff --git a/ipalib/errors.py b/ipalib/errors.py index 1c109ed66..8c1df4555 100644 --- a/ipalib/errors.py +++ b/ipalib/errors.py @@ -71,6 +71,7 @@ def check_type(name, type_, value, allow_None=False): return if type(value) is not type_: raise_TypeError(name, type_, value) + return value def check_isinstance(name, type_, value, allow_None=False): @@ -81,6 +82,7 @@ def check_isinstance(name, type_, value, allow_None=False): return if not isinstance(value, type_): raise_TypeError(name, type_, value) + return value class IPAError(Exception): diff --git a/ipalib/tests/test_errors.py b/ipalib/tests/test_errors.py index 9d08b5a70..a68b11748 100644 --- a/ipalib/tests/test_errors.py +++ b/ipalib/tests/test_errors.py @@ -71,8 +71,8 @@ def test_check_type(): value = 'How are you?' # Should pass: - f(name, str, value) - f(name, str, None, allow_None=True) + assert value is f(name, str, value) + assert None is f(name, str, None, allow_None=True) # Should raise TypeError check_TypeError(f, name, str, None) @@ -104,9 +104,9 @@ def test_check_isinstance(): value = 'How are you?' # Should pass: - f(name, str, value) - f(name, basestring, value) - f(name, str, None, allow_None=True) + assert value is f(name, str, value) + assert value is f(name, basestring, value) + assert None is f(name, str, None, allow_None=True) # Should raise TypeError check_TypeError(f, name, str, None) |