Question

我想测试类似于以下内容的函数：

def some_function_under_test(some_list_type_arg: List):
    map(some_other_function, some_list_type_arg)

对这种单元测试有什么好的和干净的方法？

我要模仿map函数

assert map_mock.called_once_with(...)

但如果函数将以这种方式编写怎么办

for i in some_list_type_arg:
    some_other_function(i)

如何独立于其实现测试此函数，即不将测试绑定到map函数？

Answer 1

你可以通过使用只调用原始函数的模拟来模拟每个元素来声明some_other_function被调用，例如：

import unittest

from mock import patch, Mock, call


def some_other_function(x):
    return 2 * x


def some_function_under_test(some_list_type_arg):
    return map(some_other_function, some_list_type_arg)


class Tests(unittest.TestCase):
    def test_thing(self):
        with patch('__main__.some_other_function', Mock(side_effect=some_other_function)) as other_mock:
            self.assertEqual(list(some_function_under_test([1, 2, 3])),
                             [2, 4, 6])
        self.assertEqual(other_mock.call_args_list,
                         [call(1), call(2), call(3)])


unittest.main()

干净的方法来测试函数是否已应用于集合的每个项目？

1 个答案: