我有一個script.py文件：# in script.pydef my_basic_function(value, c): return c(value).words()class BasicClass: def __init__(self, val): self.val = val def words(): return self.val和一個test.py文件：# in test.pyfrom mock import patchfrom script import my_basic_function, BasicClassdef third_function(self,): return "not {}".format(self.val)def fourth_function(): return "not ponies"def init_mock(self, val): self.val = val@patch('script.BasicClass.words', third_function)@patch('script.BasicClass.__init__', init_mock)def test_my_basic_function(): assert my_basic_function("ponies", BasicClass) == "not ponies"我可以pytest test.py從命令行成功運行。如果我想side_effect在 my中使用@patch，我必須做一些不同的事情：@patch('script.BasicClass.words', side_effect = fourth_function)@patch('script.BasicClass.__init__', init_mock)def test_my_basic_function(amock): assert my_basic_function("ponies", BasicClass) == "not ponies"即，我必須：添加一個test_my_basic_function我從不使用的參數(shù)。callfourth_function而不是third_function，因為我不能使用任何類實例變量。這兩種方式打補丁有什么區(qū)別？
1 回答
慕仙森
TA貢獻1827條經(jīng)驗 獲得超8個贊
您不需要為您正在做的事情打補?。ㄊ纠?A）。您正在傳遞參數(shù) c，您知道它是 BasicClass 的類，因此不需要補丁。如果您在您不希望實際發(fā)生的測試函數(shù)中引入了函數(shù)調(diào)用或?qū)ο蟪跏蓟?，您將需要一個補丁。
使用時@patch我們應該使用 kwargside_effect來模擬異常的引發(fā)，而不是調(diào)用其他函數(shù)。如果我們想模擬函數(shù)的返回值，請使用 kwarg return_value。如果我們只是想模擬一個函數(shù)，那么我們只需使用@patch沒有任何 kwargs。當我們使用補丁作為裝飾器時，我們需要將它們傳遞給函數(shù)。確實可以不用它們，但我們應該將它們與模擬函數(shù)一起使用，例如assert_called_once或assert_called_once_with以確保您的補丁按預期工作。請參見示例 B
==============示例A==================
import unittest
def my_basic_function(value, c):
return c(value).words()
class BasicClass:
def __init__(self, val):
self.val = val
def words(self):
return self.val
class TestMyBasicFunction(unittest.TestCase):
def test_my_basic_class(self):
value = my_basic_function("jeremy", BasicClass)
self.assertEqual("jeremy", value)
def test_my_basic_class_wrong(self):
value = my_basic_function("jeremy", BasicClass)
self.assertNotEqual("w33b", value)
============示例B======================
import unittest
from unittest.mock import patch
def ensure_str(value):
try:
return str(value)
except Exception:
raise TypeError
def my_basic_function(value, c):
value = ensure_str(value)
return c(value).words()
class BasicClass:
def __init__(self, val):
self.val = val
def words(self):
return self.val
class TestMyBasicFunction(unittest.TestCase):
@patch('script.ensure_str', return_value="jeremy")
def test_return_value(self, ensure_str_mock):
value = my_basic_function("jeremy", BasicClass)
ensure_str_mock.assert_called_once_with("jeremy")
self.assertEqual("jeremy", value)
@patch('script.ensure_str')
def test_no_return_value(self, ensure_str_mock):
value = my_basic_function("jeremy", BasicClass)
self.assertEqual(ensure_str_mock(), value)
@patch('script.ensure_str', side_effect=TypeError)
def test_side_effect(self, ensure_str_mock):
with self.assertRaises(TypeError):
value = my_basic_function({'apple': 'sauce'}, BasicClass)
ensure_str_mock.assert_called_once_with({'apple': 'sauce'})
添加回答
舉報
0/150
提交
取消