在python中使用模擬side_effect模擬功能

Question

我試圖在python模擬中使用side_effect功能，將對特定函數的調用完全替換為對另一個函數的調用。 現在，這兩個函數采用不同的參數集，我不確定如何將參數傳遞給side_effect。

我的示例代碼是這樣的：

import mock

def func1(arg1):
    print arg1

def func2():
    print 'original function'

@mock.patch('func2', side_effect=func1)
def func():
    func2()

func()

任何幫助，將不勝感激！

Answer 1

如果您沒有定義在不提供arg的情況下如何調用func1 ，那么沒有比...更好的回答了。

@mock.patch('func2', side_effect=lambda:func1("Hei... I don't know what you want to do!")
def func():
    func2()