gpt4 book ai didi

python - 用于在 python 中进行单元测试的局部变量,用于验证测试函数的功能

转载 作者:太空宇宙 更新时间:2023-11-03 23:38:21 26 4
gpt4 key购买 nike

我是单元测试和 python 的新手。我开始对 python 中的不同模块(用 c 开发)进行单元测试。在某些情况下,我发现函数不会返回任何值,也不会修改任何全局变量的值。

在这种情况下,我将如何根据一些局部变量值验证函数的功能。由于 local 在函数外部不可用,因此我无法验证局部变量的值。单元测试此类功能的正确方法应该是什么?

我已经通过下面的链接询问这个问题,它说我们不应该对局部变量执行单元测试。

override python function-local variable in unittest

在这里我可以看到一些方法可以用来在函数执行结束时测试局部变量。 https://coderanch.com/t/679691/engineering/test-local-variable-method-junit

有没有什么方法可以导出局部变量,以便单元测试该函数?

最佳答案

在 python 中没有“开箱即用”的解决方案,尽管我坚信在开发更高级的测试时能够模拟和检查局部变量发生了什么非常重要。我已经通过一个模拟函数并获取本地值的新类实现了这一点,希望它对您有所帮助。

import inspect
from textwrap import dedent
import re


class MockFunction:
""" Defines a Mock for functions to explore the details on their execution.
"""
def __init__(self, func):
self.func = func

def __call__(mock_instance, *args, **kwargs):
# Add locals() to function's return
code = re.sub('[\\s]return\\b', ' return locals(), ', dedent(
inspect.getsource(mock_instance.func)))
code = code + f'\nloc, ret = {mock_instance.func.__name__}(*args, **kwargs)'
loc = {'args': args, 'kwargs': kwargs}
exec(code, mock_instance.func.__globals__, loc)
# Put execution locals into mock instance
for l,v in loc['loc'].items():
setattr(mock_instance, l, v)
return loc['ret']

使用方法:

import unittest
from unittest import mock

# This is the function you would like to test. It can be defined somewhere else
def foo(param_a, param_b=10):
param_a = f'Hey {param_a}' # Local only
param_b += 20 # Local only
return 'bar'

# Define a test to validate what happens to local variables when you call that function
class SimpleTest(unittest.TestCase):

@mock.patch(f'{__name__}.foo', autospec=True, side_effect=MockFunction(foo))
def test_foo_return_and_local_params_values(self, mocked):
ret = foo('A')
self.assertEqual('Hey A', mocked.side_effect.param_a)
self.assertEqual(30, mocked.side_effect.param_b)
self.assertEqual('bar', ret)

正如我们所见,您能够使用模拟函数中的 side_effect 检查局部变量发生了什么。

关于python - 用于在 python 中进行单元测试的局部变量,用于验证测试函数的功能,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53223246/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com