Question

例如，如果我调用split方法（即some_string.split（＆＃34;：＆＃34;））有可能嘲笑这个。我想声明使用assert_called_once_with

调用split函数

Answer 1

我确认你不能这样做因为split（）是str对象的内置属性，你不能设置内置或扩展的属性，因为它们是只读的。

在尝试使用Python 2.7.10解释器之后的一些不确定的测试

>>> __builtins__.str.split
<method 'split' of 'str' objects>
>>> type(__builtins__.str.split)
<type 'method_descriptor'>

尝试使用函数

覆盖它

>>> type(lambda f:f)
<type 'function'>
>>> __builtins__.str.split = lambda f: f
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: can't set attributes of built-in/extension type 'str'

尝试使用callable（函数或方法）覆盖它

>>> type(callable)
<type 'builtin_function_or_method'>
>>> __builtins__.str.split = callable
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: can't set attributes of built-in/extension type 'str'

在这里深入了解CPython源代码[1]。这是由下面的函数列表引入的 Objects / typeobject.c 的限制。此函数检查我们是否尝试设置readonly属性并引发TypeError。

    type_setattro(PyTypeObject *type, PyObject *name, PyObject *value)
{
    if (!(type->tp_flags & Py_TPFLAGS_HEAPTYPE)) {
        PyErr_Format(
            PyExc_TypeError,
            "can't set attributes of built-in/extension type '%s'",
            type->tp_name);
        return -1;
    }
    if (PyObject_GenericSetAttr((PyObject *)type, name, value) < 0)
        return -1;
    return update_slot(type, name);
}

[1] https://hg.python.org/cpython/file/tip/Objects/typeobject.c#l3022

Answer 2

是的，它带有几个腔。就我而言，我已经在python3中成功模拟了str，因此可以断言split是通过特定输入调用的。

有两个空腔

使用补丁，我将原始的str类替换为继承自str的新类
在我测试的代码中，我不得不像str(str_val).split

这是一种方法：

class MyStr(str):
    def split(self, sep=None, maxsplit=-1)):
        expected_str = "some_input_mutated_inside_fn_before_split_called"
        self.assertEqual(self, expected_str)
        return super().split(sep=sep, maxsplit=maxsplit)

with patch('mymodule.str', new=MyStr):
    output = mymodule.function_that_calls_string_split(
        "some_input"
    )

是否可以从Python模拟字符串模块？

2 个答案: