一段python代码如何判断它是否在unittes下运行

Question

我有一个大型项目是使用Python unittest模块进行单元测试的。

我有一个控制系统行为大方面的小方法。当在UTs下运行时，我需要这个方法返回一个固定的结果，以提供一致的测试运行，但是对于我来说，为每个UTs模拟这个结果是很昂贵的。

有没有一种方法可以让这个单一的方法unittest知道，以便它在unittest下运行时可以修改其行为？

Answer 1

您可以在运行时仅为测试修改函数。例如：

模块.py

def func():
    return random.randint()

测试.py

import module

def replacement_func():
    return 4 # chosen by fair dice roll

module.func = replacement_func

# run unit tests here

现在，每当module中的代码调用func()时，它实际上会回调到您的replacement_func()。