时间戳

首页 > 解决方案 > pytest 可以从 doctest 捕获输出吗？

python - pytest 可以从 doctest 捕获输出吗？

问题描述

我正在运行pytest，--doctest-modules没有其他选择。

我有一个doctest失败的测试，我试图通过向print()底层代码添加语句来调试问题。

对于我的常规（非）测试，捕获输出正在工作并按预期显示doctest，但doctest测试没有在我的输出中显示标准pytest输出，即使我在那里print()发生了语句。

我可以将 my 重写doctest为非doctest测试，以使这些print()语句通过，但这听起来像是一个非常迂回的事情。

有什么方法可以告诉pytest我我希望它也捕获doctests 的输出吗？

$ python --version
Python 3.7.2
$ pytest --version
This is pytest version 3.10.1, imported from /home/gene/.pythonz/pythons/CPython-3.7.2/lib/python3.7/site-packages/pytest.py
setuptools registered plugins:
  pytest-xdist-1.26.1 at /home/gene/.pythonz/pythons/CPython-3.7.2/lib/python3.7/site-packages/xdist/plugin.py
  pytest-xdist-1.26.1 at /home/gene/.pythonz/pythons/CPython-3.7.2/lib/python3.7/site-packages/xdist/looponfail.py
  pytest-pythonpath-0.7.3 at /home/gene/.pythonz/pythons/CPython-3.7.2/lib/python3.7/site-packages/pytest_pythonpath.py
  pytest-mock-1.10.0 at /home/gene/.pythonz/pythons/CPython-3.7.2/lib/python3.7/site-packages/pytest_mock.py
  pytest-forked-1.0.1 at /home/gene/.pythonz/pythons/CPython-3.7.2/lib/python3.7/site-packages/pytest_forked/__init__.py
  hypothesis-4.4.1 at /home/gene/.pythonz/pythons/CPython-3.7.2/lib/python3.7/site-packages/hypothesis/extra/pytestplugin.py
  flaky-3.5.3 at /home/gene/.pythonz/pythons/CPython-3.7.2/lib/python3.7/site-packages/flaky/flaky_pytest_plugin.py

标签： pythonpytestdoctest

解决方案

将我的评论变成答案：最简单的方法是打印到 stderr，因为doctest仅捕获 stdout 进行比较。例子：

import sys


def greet(who):
    """Greet someone.

    >>> greet('world')
    'Hello world'

    >>> greet('fizz')
    'Hello fizz'

    >>> greet('buzz')
    'Hello buzz'
    """

    print('input:', who, file=sys.stderr)
    return f'Hello {who}'

运行测试：

$ pytest --doctest-modules -sv
======================================= test session starts ========================================
...
collected 1 item                                                                                   

spam.py::spam.greet input: world
input: fizz
input: buzz
PASSED

===================================== 1 passed in 0.03 seconds =====================================

推荐阅读