Python unittest模拟运行程序两次

Question

试图更多地理解 unittest.mock，但不确定为什么它会运行该程序两次。为简单起见，请考虑文件test.py 中的以下代码：

from unittest.mock import patch

class T():
    def __init__(self):
        self.setup()

    def setup(self):
        mock_testing = patch('test.testing').start()
        mock_testing.return_value = "new testing"

def testing():
    return "testing"

print("Hello")

t = T()

print("Setting up")

if testing() == "testing":
    print("old style")
elif testing() == "new testing":
    print("new style")

当我使用python test.py 运行脚本时，我得到：

Hello
Hello
Setting up
new style
Setting up
old style

为什么它会运行两次代码？即使它确实运行了两次，为什么'hello'会背靠背打印出来，应该像这样打印：

Hello
Setting up
new style
Hello
Setting up
old style

另外，我怎样才能让它只运行一次代码，模拟值为“新测试”？

Answer 1

这是因为脚本首先作为模块 __main__ 加载，但您使用 test.testing 调用 patch，因此 patch 将再次导入 test.py 作为 test 模块。由于在打印"Setting up" 之前调用了patch，因此test 模块的加载以及__main__ 模块和test 模块对"Hello" 的打印将在@ 之前完成987654333@ 由__main__ 模块打印。

如果您将__name__ 添加到print 的参数中，您将更容易看到发生了什么：

from unittest.mock import patch

class T():
    def __init__(self):
        self.setup()

    def setup(self):
        mock_testing = patch('test.testing').start()
        mock_testing.return_value = "new testing"

def testing():
    return "testing"

print(__name__, "Hello")

t = T()

print(__name__, "Setting up")

if testing() == "testing":
    print(__name__, "old style")
elif testing() == "new testing":
    print(__name__, "new style")

这个输出：

__main__ Hello
test Hello
test Setting up
test new style
__main__ Setting up
__main__ old style

为了避免这种情况，你应该给__main__.testing打补丁，这样修改后，上面的代码就会输出：

__main__ Hello
__main__ Setting up
__main__ new style