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标准输出的设置方法的内容就介绍到这里了，希望对大家的学习能带来一定的帮助。