python - 在文件顶部导入的模拟函数

Question

我正在尝试在一个项目中进行测试，但我遇到了一个奇怪的错误。

我用下面的玩具示例重现了非常相似的情况:

这是文件结构:

.
├── some_package
│   ├── __init__.py
│   └── some_file.py
└── test_mock_patch.py

"""some_package/some_file.py"""

# when I import here, the test fails
from math import floor


def some_func(a, b):
    # if I import here, the test passes
    # from math import floor
    return floor(a + b)

"""test_mock_patch.py"""

import pytest
from unittest import mock
from some_package.some_file import some_func


@pytest.fixture
def mock_floor():
    with mock.patch('math.floor', autospec=True) as m:
        yield m


def test_some_func(mock_floor):
    some_func(1.1, 1)
    assert mock_floor.call_count == 1

使用的命令:pytest -v -s test_mock_patch.py​​

错误:

为什么当我在函数内部导入时，test_some_func 通过，而当我在顶部导入时，测试失败？

预先感谢您帮助解释 mock.patch

的这种行为

版本:

• python 3.7.3
• pytest 4.4.1

Answer 1

这是一个最小的示例，说明如何通过更改 test_mock_patch.py​​ 文件来获得所需的结果。

import pytest
from some_package.some_file import some_func


def test_some_func(monkeypatch):
    with monkeypatch.context() as mc:
        mc.setattr('some_package.some_file.floor', lambda x: 'foo')
        res = some_func(1.1, 1)
        assert res == 'foo'

就像我在评论中提到的，您需要在导入函数的地方修补它。