在python中模拟一个全局变量

Question

我在主模块中有一个函数，它接受两个值并对它们执行操作。这使用了在调用此函数之前创建的全局变量

def calc(keys,values):
if globalvar == "calc":
return sum(keys)
else:
return sum(values)

正在对进行单元测试

class Testcalc(TestCase):
@mock.patch('module.globalvar ', "calc")
def test_unit(self,calc):
keys=[1,2,3]
values=[4,5,6]
sum=module.calc(keys,values)
"""
check asserts
"""

我遇到一个类型错误，参数无效。

TypeError('test_unit() takes exactly 2 arguments (1 given)',)

有人能告诉我嘲笑全局变量的正确方法吗

更新：这对我有效，不确定为什么

class Testcalc(TestCase):
@mock.patch('module.globalvar')
def test_unit(self,var):
keys=[1,2,3]
values=[4,5,6]
var="calc"
sum=module.calc(keys,values)
"""
check asserts
"""

谢谢大家

Answer 1

一个module.globalvar = 'anything'应该足够了，不需要mock.patch

def test_calc2(self):
keys = [1, 2, 3]
values = [4, 5, 6]
module.globalvar = "calc"
sum = module.calc(keys, values)
self.assertEqual(module.globalvar, 'calc')
self.assertEqual(sum, 6)
module.globalvar = 'other'
sum = module.calc(keys, values)
self.assertEqual(sum, 15)

这使用PropertyMock 工作

@mock.patch('module.globalvar', new_callable=mock.PropertyMock)
def test_calc3(self, mocked_globalvar):

这是语法正确的，但将无法通过测试，因为全局变量必须由PropertyMock 设置

@mock.patch('module.globalvar')
def test_unit(self, mock_globalvar):