Question

是否可以使用python doctest处理多行语句？

例如，以下内容在doctest中不起作用：

>>> for s in [1,2,3]:
...     for t in [4,5,6]:
...         print(s*t)

我需要从doctest执行上述三个语句。

Answer 1

你可能做错了什么。以下是正确的例子。

test.py：

"""
>>> for s in [1,2,3]:
...     for t in [4,5,6]:
...         print s*t
4
5
6
8
10
12
12
15
18
"""

它运作得很好：

$ python -m doctest -v test.py
Trying:
    for s in [1,2,3]:
        for t in [4,5,6]:
            print s*t
Expecting:
    4
    5
    6
    8
    10
    12
    12
    15
    18
ok
1 items had no tests:
    test.print_and_return
1 items passed all tests:
   1 tests in test
1 tests in 2 items.
1 passed and 0 failed.
Test passed.

另请注意doctest捕获返回值和输出：

def print_and_return():
    """
    >>> print_and_return()
    1
    2
    """
    print(1)
    return 2

任何预期的输出必须紧跟最后的＆＃39;＆gt;＆gt;＆gt; ＆＃39;或者＆＃39; ...＆＃39; 包含代码的行，以及预期的输出（如果有）扩展到下一个＆＃39;＆gt;＆gt;＆gt; ＆＃39;或所有空白行。

https://docs.python.org/2/library/doctest.html#how-it-works

Answer 2

编辑：我的回答是错误的;请参阅下面raacer的评论。我不能删除它，因为它是公认的答案。

这不是doctest的工作方式。它测试一个计算结果为值的表达式;它不捕获和测试输出。所以你想要做的是创建一个可以测试的列表，你可以使用列表理解在一行中轻松完成：

>>> [s * t for s in [1, 2, 3] for t in [4, 5, 6]]
[4, 5, 6, 8, 10, 12, 12, 15, 18]

使用Python doctest的多行语句

2 个答案: