Python如何测试stdout输出？stdout标准输出的设置方法

我们之前学习过有关Python输出的问题，使用的是priint函数进行输出，今天来学习一个新的输出方式，这篇文章主要介绍了Python如何测试stdout输出？stdout标准输出的设置方法，帮助大家更好的理解和学习Python，感兴趣的朋友可以了解下。

假设在我们的程序中，我们需要使用一个方法来进行输出，并且是以一个标准的形式，那就是使用sys.stdout，可以将文本打印到屏幕中，我们现在编写一个测试，来对它进行验证，写给定一个输入，然后再将相关的内容进行输出。解决方法，需要使用到unittest.mock 模块中的 patch() 函数，使用起来非常的简单，我们先在mymodule这个模块中，定义函数，代码如下：

def urlprint(protocol, host, domain):
  url = '{}://{}.{}'.format(protocol, host, domain)
  print(url)

在正常的情况下，一个print函数它会将输出的数据发送给sys.stdout。为了测试输出的真实性，我们可以使用一个对象来模拟它的操作，使用 unittest.mock 模块的 patch() 方法可以很方便的在测试运行的上下文中替换对象，并且当测试完成时候自动返回它们的原有状态，测试代码如下：

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 = 'xxx.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这个变量会被设置成包含期望的输出的字符串。我们还使用了with语句，它是作用就是在整个程序的执行中，对每一项操作都进行检查，当with语句结束时，patch 会将所有东西恢复到测试开始前的状态。 

有关Python如何测试stdout输出？stdout标准输出的设置方法的内容就介绍到这里了，希望对大家的学习能带来一定的帮助。

关于Python技术储备

学好 Python 不论是就业还是做副业赚钱都不错，但要学会 Python 还是要有一个学习规划。最后大家分享一份全套的 Python 学习资料，给那些想学习 Python 的小伙伴们一点帮助！

一、Python所有方向的学习路线

Python所有方向的技术点做的整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照上面的知识点去找对应的学习资源，保证自己学得较为全面。

二、Python必备开发工具

三、Python视频合集

观看零基础学习视频，看视频学习是最快捷也是最有效果的方式，跟着视频中老师的思路，从基础到深入，还是很容易入门的。

四、实战案例

光学理论是没用的，要学会跟着一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

五、Python练习题

检查学习结果。

六、面试资料

我们学习Python必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有阿里大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

 最后祝大家天天进步！！

上面这份完整版的Python全套学习资料已经上传至CSDN官方，朋友如果需要可以直接微信扫描下方CSDN官方认证二维码免费领取【保证100%免费】。