問(wèn)題

你的程序中有個(gè)方法會(huì)輸出到標(biāo)準(zhǔn)輸出中（sys.stdout）。也就是說(shuō)它會(huì)將文本打印到屏幕上面。 你想寫個(gè)測(cè)試來(lái)證明它，給定一個(gè)輸入，相應(yīng)的輸出能正常顯示出來(lái)。

解決方案

使用 unittest.mock 模塊中的 patch() 函數(shù)， 使用起來(lái)非常簡(jiǎn)單，可以為單個(gè)測(cè)試模擬 sys.stdout 然后回滾， 并且不產(chǎn)生大量的臨時(shí)變量或在測(cè)試用例直接暴露狀態(tài)變量。

作為一個(gè)例子，我們?cè)?mymodule 模塊中定義如下一個(gè)函數(shù)：

# mymodule.pydef urlprint(protocol, host, domain): url = ’{}://{}.{}’.format(protocol, host, domain) print(url)

默認(rèn)情況下內(nèi)置的 print 函數(shù)會(huì)將輸出發(fā)送到 sys.stdout 。 為了測(cè)試輸出真的在那里，你可以使用一個(gè)替身對(duì)象來(lái)模擬它，然后使用斷言來(lái)確認(rèn)結(jié)果。 使用 unittest.mock 模塊的 patch() 方法可以很方便的在測(cè)試運(yùn)行的上下文中替換對(duì)象， 并且當(dāng)測(cè)試完成時(shí)候自動(dòng)返回它們的原有狀態(tài)。下面是對(duì) mymodule 模塊的測(cè)試代碼：

from io import StringIOfrom unittest import TestCasefrom unittest.mock import patchimport mymoduleclass 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ù)，測(cè)試方法開(kāi)始會(huì)先設(shè)置每一個(gè)參數(shù)的值。 expected_url 變量被設(shè)置成包含期望的輸出的字符串。

unittest.mock.patch() 函數(shù)被用作一個(gè)上下文管理器，使用 StringIO 對(duì)象來(lái)代替 sys.stdout . fake_out 變量是在該進(jìn)程中被創(chuàng)建的模擬對(duì)象。 在with語(yǔ)句中使用它可以執(zhí)行各種檢查。當(dāng)with語(yǔ)句結(jié)束時(shí)，patch 會(huì)將所有東西恢復(fù)到測(cè)試開(kāi)始前的狀態(tài)。 有一點(diǎn)需要注意的是某些對(duì)Python的C擴(kuò)展可能會(huì)忽略掉 sys.stdout 的配置而直接寫入到標(biāo)準(zhǔn)輸出中。 限于篇幅，本節(jié)不會(huì)涉及到這方面的講解，它適用于純Python代碼。 如果你真的需要在C擴(kuò)展中捕獲I/O，你可以先打開(kāi)一個(gè)臨時(shí)文件，然后將標(biāo)準(zhǔn)輸出重定向到該文件中。

以上就是Python如何測(cè)試stdout輸出的詳細(xì)內(nèi)容，更多關(guān)于Python測(cè)試stdout輸出的資料請(qǐng)關(guān)注好吧啦網(wǎng)其它相關(guān)文章！