Question

是否可以在Python 3.6中模拟内置len()函数？

我有一个类，它定义了一个依赖于len()函数的简单方法，如下所示：

class MyLenFunc(object):
    def is_longer_than_three_characters(self, some_string):
        return len(some_string) > 3

我正在尝试为此方法编写单元测试，但我无法模拟len()函数而不会产生错误。这是我到目前为止所做的：

import unittest
from unittest.mock import patch
import my_len_func


class TestMyLenFunc(unittest.TestCase):

    @patch('builtins.len')
    def test_len_function_is_called(self, mock_len):
        # Arrange
        test_string = 'four'

        # Act
        test_object = my_len_func.MyLenFunc()
        test_object.is_longer_than_three_characters(test_string)

        # Assert
        self.assertEqual(1, mock_len.call_count)


if __name__ == '__main__':
    unittest.main()

我发现了另一个SO问题/答案here，这表明不可能模拟内置函数，因为它们是不可变的。但是，我发现了另外几个网站here和here。我在上面的单元测试课中的尝试直接取自后面这些网站（是的，我尝试过那里提到的其他技术。所有这些都以相同的错误结束）。

我得到的错误是发布完整的东西相当长，所以我会剪掉它的重复部分（你会看到它是从错误信息的最后部分递归的）。错误文本如下：

ERROR: test_len_function_is_called (__main__.TestMyLenFunc)
---------------------------------------------------------------------- 
Traceback (most recent call last):
    File "C:\Python36\Lib\unittest\mock.py", line 1179, in patched
        return func(*args, **keywargs)
    File "C:/Python/scratchpad/my_len_func_tests.py", line 16, in test_len_function_is_called
        test_object.is_longer_than_three_characters(test_string)
    File "C:\Python\scratchpad\my_len_func.py", line 3, in is_longer_than_three_characters
        return len(some_string) > 3
    File "C:\Python36\Lib\unittest\mock.py", line 939, in __call__
        return _mock_self._mock_call(*args, **kwargs)
    File "C:\Python36\Lib\unittest\mock.py", line 949, in _mock_call
        _call = _Call((args, kwargs), two=True)
    File "C:\Python36\Lib\unittest\mock.py", line 1972, in __new__
        _len = len(value)
    ...
    (Repeat errors from lines 939, 949, and 1972 another 95 times!)
    ...
    File "C:\Python36\Lib\unittest\mock.py", line 939, in __call__
        return _mock_self._mock_call(*args, **kwargs)
    File "C:\Python36\Lib\unittest\mock.py", line 944, in _mock_call
        self.called = True RecursionError: maximum recursion depth exceeded while calling a Python object

----------------------------------------------------------------------
Ran 1 test in 0.004s

FAILED (errors=1)

非常感谢提前。

Answer 1

不要修补builtins.len;现在代码在mock库中断，因为它需要正常运行的功能！修补被测模块的全局变量：

@patch('my_len_func.len')

此将全局len添加到您的模块中，len(some_string)方法中的MyLenFunc().is_longer_than_three_characters()将找到一个而不是内置函数。

但是，我必须说，如果调用len()进行测试则感觉错误。您想要检查方法是否为给定输入生成了正确的结果，len()等基本操作只是实现该目标的小型构建块，通常不会对此进行测试。

是否可以在Python 3.6中模拟内置的len（）函数？

1 个答案: