python mock验证子函数调用的参数

Question

def doublen(n):
    n = fun(n)
    return 2*n

def fun(n):
    return n+1

def test_wy(self):
   client = mock.Mock()
   client.double(1)
   client.double.assert_called_with(1)
   client.double.fun.assert_called_with(1)

第二个断言client.double.fun.assert_called_with(1)会抛出错误，我的代码有错误吗？

Answer 1

Mock只是用来模拟系统中的不同对象（创建虚拟对象）。这不是你在想什么。

如果你想看到没有错误 client.double.fun.assert_called_with（1）

然后你应该在之前调用client.double.fun（1）。