优草派  >   Python

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

翟嘉慧            来源:优草派

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

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

假设在我们的程序中,我们需要使用一个方法来进行输出,并且是以一个标准的形式,那就是使用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标准输出的设置方法的内容就介绍到这里了,希望对大家的学习能带来一定的帮助。

【原创声明】凡注明“来源:优草派”的文章,系本站原创,任何单位或个人未经本站书面授权不得转载、链接、转贴或以其他方式复制发表。否则,本站将依法追究其法律责任。
TOP 10
  • 周排行
  • 月排行