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

标签： pythontestingmockingpytest

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'

就像我在评论中提到的那样，您需要修补正在导入的功能。