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