欧美视频一,久久久久国产精品免费,国产网站在线

問題

你的程序中有個方法會輸出到標準輸出中（sys.stdout）。也就是說它會將文本打印到屏幕上面。你想寫個測試來證明它，給定一個輸入，相應的輸出能正常顯示出來。

解決方案

使用 unittest.mock 模塊中的 patch() 函數，使用起來非常簡單，可以為單個測試模擬 sys.stdout 然后回滾，并且不產生大量的臨時變量或在測試用例直接暴露狀態變量。

作為一個例子，我們在 mymodule 模塊中定義如下一個函數：

									# mymodule.py

									def urlprint(protocol, host, domain):

									  url = '{}://{}.{}'.format(protocol, host, domain)

									  print(url)

默認情況下內置的 print 函數會將輸出發送到 sys.stdout 。為了測試輸出真的在那里，你可以使用一個替身對象來模擬它，然后使用斷言來確認結果。使用 unittest.mock 模塊的 patch() 方法可以很方便的在測試運行的上下文中替換對象，并且當測試完成時候自動返回它們的原有狀態。下面是對 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() 函數接受三個參數，測試方法開始會先設置每一個參數的值。 expected_url 變量被設置成包含期望的輸出的字符串。

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

以上就是Python如何測試stdout輸出的詳細內容，更多關于Python測試stdout輸出的資料請關注服務器之家其它相關文章！

原文鏈接：https://python3-cookbook.readthedocs.io/zh_CN/latest/c14/p01_testing_output_sent_to_stdout.html