python - 是否可以模拟 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 单元测试”、“python 表达式模拟”或“python lambda 模拟” 建议我可能问错了问题。我需要模拟 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'

如果出于某种原因您不想这样做，那么您可以使用检查模块来查看 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)