使用Python doctest的多行语句

Question

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

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

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

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

Answer 1

编辑:我的回答是错误的；参见下面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]

Answer 2

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

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

任何预期的输出都必须紧跟最后的“>>>”或“.‘包含代码的行，并且预期的输出(如果有的话)扩展到下一个'>>>‘或全空格行。

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