Mocking 没有 `if __name__=="__main__":` 块的脚本

Question

假设我的根目录中有文件main.py和common.py 。 这些代表我暂时不想接触的第三方模块。 假设我还有一个test.py文件，我想用它来测试main.py 。

common.py包含以下内容：

def hello():
  print("hello")

main.py包含以下内容：

from common import hello

hello()

有没有办法在test.py中修补hello function 以便导入main.py将使用模拟的 function？ 例如，我想用猴子补丁hello打印出goodbye 。

Answer 1

很简单：

import common

common.hello = lambda: print("goodbye")

import main