Python如何测试stdout输出
问题
你的程序中有个方法会输出到标准输出中（sys.stdout）。也就是说它会将文本打印到屏幕上面。你想写个测试来证明它，给定一个输入，相应的输出能正常显示出来。
解决方案
使用unittest.mock模块中的patch()函数，使用起来非常简单，可以为单个测试模拟sys.stdout然后回滚，并且不产生大量的临时变量或在测试用例直接暴露状态变量。
作为一个例子，我们在mymodule模块中定义如下一个函数：
#mymodule.py
defurlprint(protocol,host,domain):
url='{}://{}.{}'.format(protocol,host,domain)
print(url)
默认情况下内置的print函数会将输出发送到sys.stdout。为了测试输出真的在那里，你可以使用一个替身对象来模拟它，然后使用断言来确认结果。使用unittest.mock模块的patch()方法可以很方便的在测试运行的上下文中替换对象，并且当测试完成时候自动返回它们的原有状态。下面是对mymodule模块的测试代码：
fromioimportStringIO
fromunittestimportTestCase
fromunittest.mockimportpatch
importmymodule
classTestURLPrint(TestCase):
deftest_url_gets_to_stdout(self):
protocol='http'
host='www'
domain='example.com'
expected_url='{}://{}.{}\n'.format(protocol,host,domain)
withpatch('sys.stdout',new=StringIO())asfake_out:
mymodule.urlprint(protocol,host,domain)
self.assertEqual(fake_out.getvalue(),expected_url)
讨论
urlprint()函数接受三个参数，测试方法开始会先设置每一个参数的值。expected_url变量被设置成包含期望的输出的字符串。
unittest.mock.patch()函数被用作一个上下文管理器，使用StringIO对象来代替sys.stdout.fake_out变量是在该进程中被创建的模拟对象。在with语句中使用它可以执行各种检查。当with语句结束时，patch会将所有东西恢复到测试开始前的状态。有一点需要注意的是某些对Python的C扩展可能会忽略掉sys.stdout的配置而直接写入到标准输出中。限于篇幅，本节不会涉及到这方面的讲解，它适用于纯Python代码。如果你真的需要在C扩展中捕获I/O，你可以先打开一个临时文件，然后将标准输出重定向到该文件中。
以上就是Python如何测试stdout输出的详细内容，更多关于Python测试stdout输出的资料请关注毛票票其它相关文章！
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。