python - 将python doctest放在代码文件的末尾？

Question

我可以将 python doctests 放在每个函数的主体中，有时我喜欢小型库，因为它们与函数位于同一文件中。

或者我可以将它们全部放在一个单独的文件中并执行单独的文件，这很好，以防我不希望在函数之间进行 doctest。有时我发现如果文档字符串很小，代码会更容易处理。

是否还有一种方法可以将 python doctests 保存在同一个文件中，但将它们放在文件末尾？

---

编辑:基于以下已接受答案的解决方案:

def hello_world():
  return u'Hello World'


def hello(name):
  return u'Hello %s' % name


def doctest_container():
  """
  >>> hello_world()
  u'Hello World'

  >>> hello(u'Guido')
  u'Hello Guido'
  """
  pass


if __name__ == "__main__":
    import doctest
    doctest.testmod()

其实很简单，创建一个虚拟函数作为最后一个函数，它包含一个文档字符串中的所有文档测试。

Answer 1

您可以像这样将 doctests 附加到文件末尾的文档字符串中:

def myfunc():
    """This is a docstring without a doctest
    """
    pass

# ... some other code here

# Add docstrings for doctest:
myfunc.__doc__ += """
>>> myfunc()
>>> repr(myfunc())
None
"""