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