是否可以模擬一個lambda表達式？

Question

前兩個函數display_pane_1和template_1在方法test_1中易於測試 。我想將這兩個函數重構爲一個函數display_pane_2。

lambdademo.py：

def display_pane_1(): 
    display_register(template_1) 


def template_1(): 
    return 'hello mum' 


def display_pane_2(): 
    display_register(lambda: 'hello mum') 


def display_register(template): 
    print(template()) 

test_lambdademo.py

import unittest 
import unittest.mock as mock 

import lambdademo 


class TestLambda1(unittest.TestCase): 
    def setUp(self): 
     p = mock.patch('lambdademo.display_register') 
     self.mock_display_register = p.start() 
     self.addCleanup(p.stop) 

    def test_1(self): 
     lambdademo.display_pane_1() 
     self.mock_display_register.assert_called_with(lambdademo.template_1) 

    def test_2(self): 
     lambdademo.display_pane_2() 
     self.mock_display_register.assert_called_with('????????') 

你能不能幫我寫display_pane_2有效的測試？我想測試完整的lambda表達式，即lambda x: 'hell mum'應該失敗。

我試過兩種解決方案。

第一個選項是test_1的簡單副本，用lambda的模擬替換參數lambdademo.template_1。我在手冊中找不到任何內容，暗示我應該如何嘲笑像lambda這樣的表達式。 如果是在手冊中，請告訴我在哪裏。

我的第二個選擇是在Stack Overflow和Internet上進行更廣泛的搜索。 'python expression unittest'缺少響應命中， 'python lambda unittest'，'python expression mock'或'python lambda mock' 暗示我可能會問錯誤的問題。我的假設是否需要模擬lambda表達式？

我知道一個簡單的編碼解決方案將保留原始代碼，但在這一點上我更感興趣填補我的知識空白。

Answer 1

如果lambda表達式可以在類似某個類或模塊的屬性的地方訪問，那麼你可以嘲笑它，但這似乎不太可能。通常，當你不需要對函數的引用時，會使用lambda表達式。否則，您只需使用常規功能。

但是，您可以檢索模擬對象上所有調用的參數，以便查看傳入的lambda表達式。在您給出的示例中，最簡單的方法是調用lambda表達式並查看它返回的內容。

from mock import patch 

def foo(bar): 
    return bar() 

def baz(): 
    return 42 

print foo(baz) 

with patch('__main__.foo') as mock_foo: 
    print foo(baz) 
    print foo(lambda: 'six by nine') 

    assert mock_foo.call_args_list[0][0][0]() == 42 
    assert mock_foo.call_args_list[1][0][0]() == 'six by nine' 

如果由於某種原因您不想這樣做，那麼您可以使用inspect模塊來查看lambda表達式。下面是剛剛轉儲，被定義功能的源代碼行的例子：

from inspect import getsource 
from mock import patch 

def foo(bar): 
    return bar() 

def baz(): 
    return 42 

print foo(baz) 

with patch('__main__.foo') as mock_foo: 
    print foo(baz) 
    print foo(lambda: 'six by nine') 
    print mock_foo.call_args_list 
    for call_args in mock_foo.call_args_list: 
     print '---' 
     print getsource(call_args[0][0]) 

結果：

42 
<MagicMock name='foo()' id='140595519812048'> 
<MagicMock name='foo()' id='140595519812048'> 
[call(<function baz at 0x7fdef208fc08>), 
call(<function <lambda> at 0x7fdef208fe60>)] 
--- 
def baz(): 
    return 42 

--- 
    print foo(lambda: 'six by nine') 

下面是一個版本的測試，以你的示例代碼通過。它測試兩種方式：調用模板並檢查模板的來源。

# test_lambdademo.py 

from inspect import getsource 
import unittest 
import unittest.mock as mock 

import lambdademo 


class TestLambda1(unittest.TestCase): 
    def setUp(self): 
     p = mock.patch('lambdademo.display_register') 
     self.mock_display_register = p.start() 
     self.addCleanup(p.stop) 

    def test_1(self): 
     lambdademo.display_pane_1() 
     self.mock_display_register.assert_called_with(lambdademo.template_1) 

    def test_2(self): 
     lambdademo.display_pane_2() 
     template = self.mock_display_register.call_args[0][0] 
     template_content = template() 
     template_source = getsource(template) 

     self.assertEqual('hello mum', template_content) 
     self.assertIn('hello mum', template_source)