python单元测试断言用部分调用_python

使用 python 3.5.3 我想断言模拟函数收到了特定的参数，但我不想检查所有收到的参数，只检查对测试很重要的参数。

例如，而不是这样做：

my_func_mock.assert_called_with('arg1','arg2','arg3')

我想做类似的事情：

my_func_mock.assert_called_with_partial(arg2='arg2')

是否可以？

assert_called_with检查最后一次调用是如何进行的。 在 MagicMock 对象中，可以在其call_args.args属性中找到最后一次调用的 args。 最后一次调用的 kwargs 可以在其call_args.kwargs属性中找到。 我们可以单独检查这些。

假设在测试中我们调用了my_func_mock

my_func_mock(arg0, arg1, kwarg1 = 'kw1', kwarg2 = 'kw2')

我们可以通过以下方式直接检查单个参数

assert my_func_mock.call_args.args[0] == arg0
assert my_func_mock.call_args.args[1] == arg1

对于任何 kwargs，我们可以通过以下方式直接单独检查它们

assert my_func_mock.call_args.kwargs['kwarg1'] == 'kw1'
assert my_func_mock.call_args.kwargs['kwarg2'] == 'kw2'

您可以复制mock.NonCallableMock.assert_called_with的mock.NonCallableMock.assert_called_with并对其进行修改，以便您可以实现以下部分匹配预期参数和关键字参数的assert_called_with_partial ：

from unittest import mock

def assert_called_with_partial(_mock_self, *args, **kwargs):
    self = _mock_self
    if self.call_args is None:
        expected = self._format_mock_call_signature(args, kwargs)
        raise AssertionError('Expected call: %s\nNot called' % (expected,))

    def _error_message():
        msg = self._format_mock_failure_message(args, kwargs)
        return msg

    expected = self._call_matcher((args, kwargs))
    expected_args, expected_kwargs = expected
    actual_args, actual_kwargs = self._call_matcher(self.call_args)
    if actual_args[:len(expected_args)] != expected_args or not (expected_kwargs.items() <= actual_kwargs.items()):
        cause = expected if isinstance(expected, Exception) else None
        raise AssertionError(_error_message()) from cause

mock.NonCallableMock.assert_called_with_partial = assert_called_with_partial

以便：

m = mock.MagicMock()
m(1, 2, a=1, b=2)
m.assert_called_with_partial(1, a=1)

不会引发AssertionError ，但是：

m.assert_called_with_partial(1, 2, a=2)

将提高：

AssertionError: Expected call: mock(1, 2, a=2)
Actual call: mock(1, 2, a=1, b=2)

您可以使用 unittest.mock 中的ANY 。

from unittest.mock import MagicMock, ANY
my_func_mock = MagicMock()
my_func_mock('argument', 1, foo='keyword argument', bar={'a thing': 2})
my_func_mock.assert_called_once_with(
    'argument',
    1,
    foo='keyword argument',
    bar=ANY
)

这需要已经知道参数名称，但允许该参数的任何值。

请参阅： :

python单元测试断言用部分调用

问题描述

1楼

2楼

3楼