python - Doctest 和第三方装饰器

Question

我需要对一个必须用装饰器包装的方法进行文档测试，该装饰器不将 @functools.wraps 或 functools.update_wrapper 应用于被包装的方法.在这种情况下，doctest 看不到要测试的方法的文档字符串:

@third_party_decorator
def method_to_be_tested():
    """
    >>> method_to_be_tested()
    "foo"
    """
    return "foo"

This question类似，但是，我无法更 retrofit 饰器的代码。

我能做的最好的事情是什么？

Answer 1

您可能无法更改 third_party_decorator 的代码，但您可以将其再次包装以解决问题，甚至在必要时重新使用相同的名称:

def third_party_decorator(f):
    # badly behaved decorator, this removed f's docstring
    def no_docstring():
        return f()
    return no_docstring

old_third_party_decorator = third_party_decorator

def third_party_decorator(f):
    # replace third_party_decorator AND preserve docstring
    new_f = old_third_party_decorator(f)
    new_f.__doc__ = f.__doc__
    return new_f

@third_party_decorator
def method_to_be_tested():
    """
    >>> method_to_be_tested()
    'foo'
    """
    return "foo"

import doctest
print(doctest.testmod())