python - 模拟范围超出当前测试

Question

我正在模拟一个模块...这是我的示例代码

def test_validate(self):
    """Test Base Retriever Dataframe"""

    sampleQuoteClass = self.sampleQuoteClass('ThisQuote')

    bRet._getAsOfData = MagicMock(return_value=sampleQuoteClass)

    dataAsDataFrame = bVal.validate(metaDataName='MyNewQuote')
    self.assertTrue(len(dataAsDataFrame) > 0)

这工作正常。

问题是 - bRet._getAsOfData 也被模拟用于下一个测试，它顺便驻留在其他测试类中。

只有当所有测试作为集合的一部分一起运行时才会出现此问题。

Answer 1

听起来你可能想要 patch对象而不是直接模拟它。您可能需要稍微调整我的示例以适合您的代码，但请尝试如下操作:

from mock import patch

def test_validate(self):
    """Test Base Retriever Dataframe"""

    sampleQuoteClass = self.sampleQuoteClass('ThisQuote')

    with patch('__main__.bRet') as mock_bRet:
        mock_bRet._getAsOfData.return_value = sampleQuoteClass

        dataAsDataFrame = bVal.validate(metaDataName='MyNewQuote')
        self.assertTrue(len(dataAsDataFrame) > 0)

当您修补对象时，模拟将被撤消，一旦 with block 退出，对象将“恢复正常”，因此模拟状态不会转移到您的其他测试。也可以将 patch 用作装饰器，但我一直更喜欢将它用作上下文管理器。有关每种用法的示例，请参阅上面链接的文档。

此外，根据我的经验，打补丁可能很棘手，所以我建议您阅读 this useful bit of documentation on "where to patch"以及。