如何使用 doctest 检查程序是否产生了某些输出？

Question

在我的一个函数中，我使用 subprocess.check_call 调用外部程序，这将产生输出。我如何使用 doctest 来确保它产生的输出是我期望的输出？

Answer 1

也许这可以帮助：

import sys
import tempfile
import subprocess


def example(output):
    r""" Do something ...

    >>> output = example('Processing file ...')
    >>> print output # doctest:+ELLIPSIS
    'Processing file ...'

    Check how many file was processed.
    >>> [line.startswith('Processing file')
    ... for line in output.splitlines()].count(True)
    1    

    """
    cmd = "print '%s'" % (output, )
    with tempfile.TemporaryFile() as output:
        subprocess.check_call([sys.executable, '-c', cmd], stdout=output)
        output.seek(0)
        res = output.read()

    return res

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

正如你所看到的，我使用了 subprocess.check_call 函数的参数 stdout ，以便能够获取命令的输出，除此之外，如果你没有使用 stdout 参数（我认为这是你的情况）我认为很难捕获命令输出。

希望这是有希望的:)