Python如何测试stdout输出
作者:D 时间:2023-10-22 23:59:56
问题
你的程序中有个方法会输出到标准输出中(sys.stdout)。也就是说它会将文本打印到屏幕上面。 你想写个测试来证明它,给定一个输入,相应的输出能正常显示出来。
解决方案
使用 unittest.mock 模块中的 patch() 函数, 使用起来非常简单,可以为单个测试模拟 sys.stdout 然后回滚, 并且不产生大量的临时变量或在测试用例直接暴露状态变量。
作为一个例子,我们在 mymodule 模块中定义如下一个函数:
# mymodule.py
def urlprint(protocol, host, domain):
url = '{}://{}.{}'.format(protocol, host, domain)
print(url)
默认情况下内置的 print 函数会将输出发送到 sys.stdout 。 为了测试输出真的在那里,你可以使用一个替身对象来模拟它,然后使用断言来确认结果。 使用 unittest.mock 模块的 patch() 方法可以很方便的在测试运行的上下文中替换对象, 并且当测试完成时候自动返回它们的原有状态。下面是对 mymodule 模块的测试代码:
from io import StringIO
from unittest import TestCase
from unittest.mock import patch
import mymodule
class TestURLPrint(TestCase):
def test_url_gets_to_stdout(self):
protocol = 'http'
host = 'www'
domain = 'example.com'
expected_url = '{}://{}.{}\n'.format(protocol, host, domain)
with patch('sys.stdout', new=StringIO()) as fake_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,你可以先打开一个临时文件,然后将标准输出重定向到该文件中。
来源:https://python3-cookbook.readthedocs.io/zh_CN/latest/c14/p01_testing_output_sent_to_stdout.html
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
python遍历文件夹,指定遍历深度与忽略目录的方法
基于pycharm实现批量修改变量名
![](https://img.aspxhome.com/file/2023/1/131011_0s.png)
用Dreamweaver MX制作文字特效
![](https://img.aspxhome.com/file/UploadPic/20072/20072311216835s.gif)
自己搭建resnet18网络并加载torchvision自带权重的操作
Python Dict找出value大于某值或key大于某值的所有项方式
python 批量将中文名转换为拼音
![](https://img.aspxhome.com/file/2023/4/68354_0s.png)
python3 简单实现组合设计模式
![](https://img.aspxhome.com/file/2023/2/59482_0s.png)
Django打印出在数据库中执行的语句问题
在Go语言程序中使用gojson来解析JSON格式文件
python如何对数组进行降维
![](https://img.aspxhome.com/file/2023/0/112940_0s.png)
Python实现多线程的两种方式分析
分支任务:从哪里来,回哪里去
![](https://img.aspxhome.com/file/UploadPic/20099/4/3819638990_da91f304b0-56s.jpg)
Python3.5集合及其常见运算实例详解
python数据结构之二叉树的统计与转换实例
![](https://img.aspxhome.com/file/2023/2/61712_0s.jpg)
详解Tensorflow数据读取有三种方式(next_batch)
![](https://img.aspxhome.com/file/2023/1/71461_0s.jpg)
Python实现读取及写入csv文件的方法示例
![](https://img.aspxhome.com/file/2023/8/65528_0s.png)
MSSQL经典语句
一行CSS代码为网站加上奥运主题
![](https://img.aspxhome.com/file/UploadPic/20087/20/2008720123826347s.png)