Python中的模拟stdout

Question

我是Python单元测试的新手，尤其是Mock。我该如何模仿一个可以用来做以下事情的对象？我只需要一个不会使循环崩溃的对象，以便完成测试。在

for ln in theMock.stdout.readlines()

我试着用

^{pr2}$

以及

Mock(stdout=Mock(spec=file, wraps=StringIO())

但是它说list对象没有属性stdout。在

Answer 1

这个怎么样？在

from mock import Mock

readlines = Mock(return_value=[])
stdout = Mock(readlines=readlines)
theMock = Mock(stdout=stdout)
print(theMock.stdout.readlines())

输出：

^{pr2}$

您的for循环将被跳过，因为readlines()将返回一个空列表。在