3

我正在使用模拟库和 unittest2 来测试我的软件项目的不同方面。

目前我有以下问题:是否可以模拟一个函数,以便默认关键字参数不同,但功能仍然存在?

说我有以下代码

class C():
  def fun(self, bool_arg = True):
    if bool_arg:
      return True
    else
      return False

如果我想模拟 C.fun 怎么办:

C.fun = mock.Mock(???)

这样 C 的每个实例都会将关键字“bool_arg”替换为 False,而不是 True,结果为:

c = C()
c.fun()

返回:

错误的

4

2 回答 2

3

您也可以尝试包装您的功能。线上的东西

def wrapper(func, bool_arg):
    def inner(*args, **kwargs):
        kwargs['bool_arg']=bool_arg
        return func(*args, **kwargs)
    return inner

class C():
    def fun(...):
        ...

c = C()
c.fun = wrapper(fun, False)

应该管用

编辑

如果要更改类而不是特定实例的默认值,可以创建派生类并重新定义fun包装C. 在线上的东西(我现在没有时间测试它):

class D(C):
    def fun(self, *args, **kwargs):
        f = wrapper(C.f, False)
        return f(*args, **kwargs)

然后关于@Ber的建议,你可以定义def wrapper(func, **wrapkwargs)然后代替kwargs['bool_arg']=bool_argdo

for i in wrapkwargs.iteritems():  #wrapkwargs is a dictionary
    kwargs[i[0]] = i[1]
于 2013-01-30T14:20:21.293 回答
2

您可以尝试使用此代码:

>>> import mock
>>> 
>>> class C():
...   def fun(self, bool_arg = True):
...     if bool_arg:
...       print "True"
...     else:
...       print "False"
... 
>>> c = C()
>>> funCopy = c.fun
>>> c.fun = mock.Mock(side_effect=lambda bool_arg=False: funCopy(bool_arg=bool_arg))
>>> c.fun()
False

希望这可以帮助

于 2013-01-30T13:53:55.453 回答