覆盖模拟对象的“is”运算符

Question

简化代码：我有一个函数，它需要一个数字或 None，如果是 None，则返回 True，如果是数字，则返回 False，例如：

def function(var):
    return var is None

我想传递一个模拟对象并使用is 运算符对其进行测试，如下所示：

from unittest.mock import Mock

mock = Mock(return_value=None)
assert function()

尽管我将值嘲笑为None，但这将失败。根据documentation，支持的魔术方法有很多，但没有一个让我实现这种情况下的行为。

更改为assert mock == None 可以让我使用__eq__ 运算符，但我不想接触代码（根据this answer，显然使用is 比== 快）

Answer 1

也许使用isinstance() 而不是is 对你有用。将模拟对象类声明为 class MockInt(int): 将使 isinstance() 将变量识别为 int：

class MockInt(int):
      def __init__(self,value): self.value = value

i = MockInt(23)

print( isinstance(i,int) ) # True 

Answer 2

正如许多人所指出的，is 是“不可覆盖的”，因为它是对 id 的快速相等检查。这就是与 __eq__ 相比存在性能差异的原因。

在the docs 和this answer 的 cmets 中了解更多信息。

Answer 3

TL;DR

当有return_value时不要忘记括号

from unittest.mock import Mock

mock = Mock(return_value=None)
assert function(mock())

更多信息

如果您想传递一个包含None 的变量来测试function，则无需使用Mock。您可以编写以下代码：

mock = None
assert function(mock)

Mock 的最佳用法是当您想要模拟具有更复杂结构的方法或对象时。因此，当您编写 mock = Mock(return_value=None) 时，您必须调用 mock 以返回 return_value。