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设置globalvar，因此测试将失败。

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

在python中模拟全局变量

1 个答案: