一种简单的方法来模拟松散定义的Python dict对象

时间:2013-10-11 20:42:53

标签: python unit-testing python-mock python-unittest

有没有一种简单的方法可以在Python中模拟松散定义的dict对象?例如,如何在给定dict input的情况下轻松表达,我想检查其中的每个值是否符合特定的元定义,如最小值和最大值,长度和类型?

能够做到这一点可能很方便,例如,在编写测试时。

mock(Python版本3.3+中的unittest.mock)中,可以指定值可以是ANY值,例如

>>> mock = Mock(return_value=None)
>>> mock('foo', bar=object())
>>> mock.assert_called_once_with('foo', bar=ANY)

然而,如果上面的bar应该是类似dict的对象,比如

>>> {'baz': <an integer between -3 and 14>, 'qux': <'yes' or 'no'>}

1 个答案:

答案 0 :(得分:5)

我实际上已经编写了AnyValid,这是一个利用formencodeunittest.mock中实施的出色工作的最小库来处理此类案例。

例如,测试如上所述的dict对象可以表示为

>>> from mock import Mock
>>> from any_valid import AnyValid, Int, OneOf
>>> valid_bar = {
...     'baz': AnyValid(Int(min=-3, max=14)),
...     'qux': AnyValid(OneOf(['yes', 'no'])),
...     }
>>> mock = Mock(return_value=None)
>>> mock('foo', bar={'baz': 4, 'qux': 'yes'})
>>> mock.assert_called_once_with('foo', bar=valid_bar)
>>>

因为AnyValid可以从formencode中的大量验证器中获取任何验证器,所以可以用类似的表达方式指定许多其他条件。