亚洲精品久久久中文字幕-亚洲精品久久片久久-亚洲精品久久青草-亚洲精品久久婷婷爱久久婷婷-亚洲精品久久午夜香蕉

您的位置:首頁技術文章
文章詳情頁

Python如何在單元測試中給對象打補丁

瀏覽:3日期:2022-07-15 11:49:37

問題

你寫的單元測試中需要給指定的對象打補丁, 用來斷言它們在測試中的期望行為(比如,斷言被調用時的參數個數,訪問指定的屬性等)。

解決方案

unittest.mock.patch() 函數可被用來解決這個問題。 patch() 還可被用作一個裝飾器、上下文管理器或單獨使用,盡管并不常見。 例如,下面是一個將它當做裝飾器使用的例子:

from unittest.mock import patchimport example@patch(’example.func’)def test1(x, mock_func): example.func(x) # Uses patched example.func mock_func.assert_called_with(x)

它還可以被當做一個上下文管理器:

with patch(’example.func’) as mock_func: example.func(x) # Uses patched example.func mock_func.assert_called_with(x)

最后,你還可以手動的使用它打補丁:

p = patch(’example.func’)mock_func = p.start()example.func(x)mock_func.assert_called_with(x)p.stop()

如果可能的話,你能夠疊加裝飾器和上下文管理器來給多個對象打補丁。例如:

@patch(’example.func1’)@patch(’example.func2’)@patch(’example.func3’)def test1(mock1, mock2, mock3): ...def test2(): with patch(’example.patch1’) as mock1, patch(’example.patch2’) as mock2, patch(’example.patch3’) as mock3: ...

討論

patch() 接受一個已存在對象的全路徑名,將其替換為一個新的值。 原來的值會在裝飾器函數或上下文管理器完成后自動恢復回來。 默認情況下,所有值會被 MagicMock 實例替代。例如:

>>> x = 42>>> with patch(’__main__.x’):... print(x)...<MagicMock name=’x’ id=’4314230032’>>>> x42>>>

不過,你可以通過給 patch() 提供第二個參數來將值替換成任何你想要的:

>>> x42>>> with patch(’__main__.x’, ’patched_value’):... print(x)...patched_value>>> x42>>>

被用來作為替換值的 MagicMock 實例能夠模擬可調用對象和實例。 他們記錄對象的使用信息并允許你執行斷言檢查,例如:

>>> from unittest.mock import MagicMock>>> m = MagicMock(return_value = 10)>>> m(1, 2, debug=True)10>>> m.assert_called_with(1, 2, debug=True)>>> m.assert_called_with(1, 2)Traceback (most recent call last): File '<stdin>', line 1, in <module> File '.../unittest/mock.py', line 726, in assert_called_with raise AssertionError(msg)AssertionError: Expected call: mock(1, 2)Actual call: mock(1, 2, debug=True)>>>>>> m.upper.return_value = ’HELLO’>>> m.upper(’hello’)’HELLO’>>> assert m.upper.called>>> m.split.return_value = [’hello’, ’world’]>>> m.split(’hello world’)[’hello’, ’world’]>>> m.split.assert_called_with(’hello world’)>>>>>> m[’blah’]<MagicMock name=’mock.__getitem__()’ id=’4314412048’>>>> m.__getitem__.calledTrue>>> m.__getitem__.assert_called_with(’blah’)>>>

一般來講,這些操作會在一個單元測試中完成。例如,假設你已經有了像下面這樣的函數:

# example.pyfrom urllib.request import urlopenimport csvdef dowprices(): u = urlopen(’http://finance.yahoo.com/d/quotes.csv?s=@^DJI&f=sl1’) lines = (line.decode(’utf-8’) for line in u) rows = (row for row in csv.reader(lines) if len(row) == 2) prices = { name:float(price) for name, price in rows } return prices

正常來講,這個函數會使用 urlopen() 從Web上面獲取數據并解析它。 在單元測試中,你可以給它一個預先定義好的數據集。下面是使用補丁操作的例子:

import unittestfrom unittest.mock import patchimport ioimport examplesample_data = io.BytesIO(b’’’'IBM',91.1r'AA',13.25r'MSFT',27.72rr’’’)class Tests(unittest.TestCase): @patch(’example.urlopen’, return_value=sample_data) def test_dowprices(self, mock_urlopen): p = example.dowprices() self.assertTrue(mock_urlopen.called) self.assertEqual(p, {’IBM’: 91.1, ’AA’: 13.25, ’MSFT’ : 27.72})if __name__ == ’__main__’: unittest.main()

本例中,位于 example 模塊中的 urlopen() 函數被一個模擬對象替代, 該對象會返回一個包含測試數據的 ByteIO()

還有一點,在打補丁時我們使用了 example.urlopen 來代替 urllib.request.urlopen 。 當你創建補丁的時候,你必須使用它們在測試代碼中的名稱。 由于測試代碼使用了 from urllib.request import urlopen ,那么 dowprices() 函數 中使用的 urlopen() 函數實際上就位于 example 模塊了。

本節實際上只是對 unittest.mock 模塊的一次淺嘗輒止。 更多更高級的特性,請參考 官方文檔

以上就是Python如何在單元測試中給對象打補丁的詳細內容,更多關于Python 單元測試的資料請關注好吧啦網其它相關文章!

標簽: Python 編程
相關文章:
主站蜘蛛池模板: 另类图片第一页 | 欧美性色黄大片一级毛片视频 | 色综合合久久天天给综看 | 在线观看扣喷水 | 国产丝袜制服 | 西西人体午夜视频 | 久久综合精品不卡一区二区 | 欧美日韩亚洲高清不卡一区二区三区 | 五月激激激综合网色播免费 | 日韩毛片在线看 | 国产在线看不卡一区二区 | 中国美女一级黄色片 | 日韩欧美在线观看一区 | 欧美一级毛片不卡免费观看 | 俄国特级毛片www免 俄罗斯14一18处交 | 日韩欧美一区二区久久 | 国产剧情网站 | 亚洲欧洲一二三区机械有限公司 | 婷婷免费在线 | 国产拍拍视频 | 黄色精品 | 国产在线精品视频 | videosg最新欧美另类 | 国产精品亚洲精品影院 | ak福利午夜在线观看 | 国产精品18久久久久久小说 | 天天综合久久 | 91免费精品国偷自产在线在线 | 2022国产成人福利精品视频 | 欧美人七十二式性视频教程一 | 国产一级做a爰片久久毛片男 | 青青草精品在线视频 | 一区二区三区免费在线 | 中文字幕在线播放 | 国产麻豆精品在线观看 | 激情九月婷婷 | 可以看黄色的网站 | 日韩精品第一区 | 日韩区在线观看 | 国产日韩欧美二区 | 特级一级毛片 |