Python如何測試stdout輸出
問題
你的程序中有個方法會輸出到標準輸出中（sys.stdout）。也就是說它會將文本打印到屏幕上面。 你想寫個測試來證明它，給定一個輸入，相應(yīng)的輸出能正常顯示出來。
解決方案
使用 unittest.mock 模塊中的 patch() 函數(shù)， 使用起來非常簡單，可以為單個測試模擬 sys.stdout 然后回滾， 并且不產(chǎn)生大量的臨時變量或在測試用例直接暴露狀態(tài)變量。
作為一個例子，我們在 mymodule 模塊中定義如下一個函數(shù)：
# mymodule.py
def urlprint(protocol, host, domain):
url = '{}://{}.{}'.format(protocol, host, domain)
print(url)
默認情況下內(nèi)置的 print 函數(shù)會將輸出發(fā)送到 sys.stdout 。 為了測試輸出真的在那里，你可以使用一個替身對象來模擬它，然后使用斷言來確認結(jié)果。 使用 unittest.mock 模塊的 patch() 方法可以很方便的在測試運行的上下文中替換對象， 并且當測試完成時候自動返回它們的原有狀態(tài)。下面是對 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() 函數(shù)接受三個參數(shù)，測試方法開始會先設(shè)置每一個參數(shù)的值。 expected_url 變量被設(shè)置成包含期望的輸出的字符串。
unittest.mock.patch() 函數(shù)被用作一個上下文管理器，使用 StringIO 對象來代替 sys.stdout . fake_out 變量是在該進程中被創(chuàng)建的模擬對象。 在with語句中使用它可以執(zhí)行各種檢查。當with語句結(jié)束時，patch 會將所有東西恢復到測試開始前的狀態(tài)。 有一點需要注意的是某些對Python的C擴展可能會忽略掉 sys.stdout 的配置而直接寫入到標準輸出中。 限于篇幅，本節(jié)不會涉及到這方面的講解，它適用于純Python代碼。 如果你真的需要在C擴展中捕獲I/O，你可以先打開一個臨時文件，然后將標準輸出重定向到該文件中。
以上就是Python如何測試stdout輸出的詳細內(nèi)容，更多關(guān)于Python測試stdout輸出的資料請關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
今天小編就為大家分享一篇opencv3/C++實現(xiàn)視頻讀取、視頻寫入，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-12-12
JWT的聲明一般被用來在身份提供者和服務(wù)提供者間傳遞被認證的用戶身份信息，以便于從資源服務(wù)器獲取資源，也增加一些額外的其它業(yè)務(wù)邏輯所必須的聲明信息，該token也可直接被用于認證，也可被加密,這篇文章主要介紹了Python JWT認證與pyjwt包簡介,需要的朋友可以參考下2023-05-05
python 實現(xiàn)圍棋游戲(純tkinter gui)
這篇文章主要介紹了python 如何實現(xiàn)圍棋游戲，幫助大家利用tkinter制作圖形界面程序，感興趣的朋友可以了解下2020-11-11
Python?pygame項目實戰(zhàn)英雄動畫特效實現(xiàn)
這篇文章主要為大家介紹了Python?pygame項目實戰(zhàn)英雄動畫特效實現(xiàn)示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-08-08
在C#中很容易生成一組唯一碼，最常用的是結(jié)構(gòu)體GUID的NewGuid()實例。如果C#運行Guid.NewGuid();將會得到據(jù)說世界唯一的號碼，形如：887687be-00cf-4dca-8fe4-7c4fc19b9ecc 。最近看了一下Python的相關(guān)模塊，也發(fā)現(xiàn)了一個模塊uuid。下面來看看詳細的介紹與使用示例吧。2016-12-12
pyinstaller打包opencv和numpy程序運行錯誤解決
這篇文章主要介紹了pyinstaller打包opencv和numpy程序運行錯誤解決,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2019-08-08
Python OpenCV高斯金字塔與拉普拉斯金字塔的實現(xiàn)
這篇文章主要介紹了Python OpenCV高斯金字塔與拉普拉斯金字塔的實現(xiàn)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-03-03
Django博客系統(tǒng)注冊之創(chuàng)建用戶模塊應(yīng)用
本文主要介紹了Django博客系統(tǒng)注冊之創(chuàng)建用戶模塊應(yīng)用，文中通過示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-09-09
這篇文章主要介紹了Python深拷貝與淺拷貝引用，Python并沒有拷貝這個對象，而只是拷貝了這個對象的引用，下文具體的相關(guān)介紹需要的小伙伴可以參考一下2022-04-04
這篇文章主要為大家詳細介紹了利用aardio給python編寫圖形界面的方法，具有一定的參考價值，感興趣的小伙伴們可以參考一下2017-08-08