pytestmock从modu模拟一个函数

2024-05-17 02:53:54 发布

您现在位置:Python中文网/ 问答频道 /正文

我的模块engine.py中有一个util,它是从另一个文件导入的:

from main.utils.string import get_random_string

def generate_random_string():
    return get_random_string()

在我的测试文件中:

def test_generate_random_string(mocker):
    mocker.patch('main.utils.string.get_random_string', return_value='123456')

但是,它仍然试图使用string.get_random_string的实际实现,而不是我创建的模拟,除非我将engine.py更改为:

from main.utils import string

def generate_random_string():
    return string.get_random_string()

如何在不将整个string模块导入engine.py的情况下实现模拟部分?


Tags: 模块文件frompyimportgetstringreturn