From 43d37b875fc6d218d215748a0ed62491366d85fe Mon Sep 17 00:00:00 2001 From: maxkoryukov Date: Sat, 12 Mar 2016 18:29:27 +0500 Subject: [PATCH] FIX: unittestcompat now support tests without python-doc. That was an ugly bug... --- lib/unittestcompat.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lib/unittestcompat.py b/lib/unittestcompat.py index 12a497be..1fe45a2a 100644 --- a/lib/unittestcompat.py +++ b/lib/unittestcompat.py @@ -107,7 +107,8 @@ def TestArgs(*parameters): name_for_parameter = method.__name__ + "(" + args_for_parameter + ")" frame = sys._getframe(1) # pylint: disable-msg=W0212 frame.f_locals[name_for_parameter] = method_for_parameter - frame.f_locals[name_for_parameter].__doc__ = method.__doc__ + '(' + args_for_parameter + ')' + if method.__doc__: + frame.f_locals[name_for_parameter].__doc__ = method.__doc__ + '(' + args_for_parameter + ')' method_for_parameter.__name__ = name_for_parameter + '(' + args_for_parameter + ')' return None return decorator