
在python中,函数经常使用默认参数来提供灵活性。例如,考虑一个获取用户信息的函数:
def get_user(user_id, adults_only=False):
"""
根据user_id获取用户信息。
:param user_id: 用户ID
:param adults_only: 如果为True,则只返回成年用户;默认为False。
"""
# 实际实现可能涉及数据库查询等
if user_id == 1 and adults_only:
return "adult_user_data"
elif user_id == 1 and not adults_only:
return "child_user_data" # 假设ID为1的用户在默认情况下被视为“child”
elif user_id == 2 and adults_only:
return "adult_user_data"
else:
return "unknown"当使用unittest和parameterized库进行参数化测试时,我们通常会遇到一个问题:如何在一个参数化测试中同时覆盖显式传入参数和使用默认参数的情况?
传统的做法是为默认参数情况单独编写一个测试用例,例如:
import unittest
from parameterized import parameterized
# 假设get_user函数和Child对象已定义
# 为了示例完整性,这里模拟一个Child对象
class MockChild:
def __init__(self, id):
self.id = id
# 模拟get_user函数,与上面定义的一致
def get_user(user_id, adults_only=False):
if user_id == 1 and adults_only:
return "none" # 对应示例中的"none"
elif user_id == 1 and not adults_only:
return "child" # 对应示例中的"child"
else:
return "unknown"
class TestGetUserTraditional(unittest.TestCase):
def setUp(self):
self.child = MockChild(1) # 假设child.id为1
@parameterized.expand([
(True, "none"),
(False, "child")
])
def test_get_child(self, adults_only_param, expected):
"""测试显式传入adults_only参数的情况。"""
actual = get_user(self.child.id, adults_only_param)
self.assertEqual(expected, actual)
def test_get_child_default(self):
"""单独测试使用默认参数的情况。"""
actual = get_user(self.child.id) # 不传入adults_only
self.assertEqual("child", actual)这种方法虽然有效,但当测试用例增多时,会导致代码冗余,且不利于维护。理想情况下,我们希望能够将这两种测试场景合并到一个参数化测试中。
为了将默认参数的测试合并到参数化测试中,我们可以采用一种策略:在参数化数据中引入一个“哨兵值”(sentinel value),用于标识应该使用默认参数的情况。当检测到这个哨兵值时,我们不向函数传递该参数,从而触发其默认行为。
立即学习“Python免费学习笔记(深入)”;
通常,None是一个很好的哨兵值选择,因为它在很多情况下不会作为实际业务参数被传入。然后,我们利用Python的**kwargs语法来动态构建传递给函数的关键字参数。
以下是具体的实现:
import unittest
from parameterized import parameterized
# 模拟get_user函数和MockChild对象,与上面定义的一致
class MockChild:
def __init__(self, id):
self.id = id
def get_user(user_id, adults_only=False):
if user_id == 1 and adults_only:
return "none"
elif user_id == 1 and not adults_only:
return "child"
else:
return "unknown"
class TestGetUserOptimized(unittest.TestCase):
def setUp(self):
self.child = MockChild(1) # 假设child.id为1
@parameterized.expand([
(True, "none"), # 显式传入adults_only=True
(False, "child"), # 显式传入adults_only=False
(None, "child") # 使用None作为哨兵值,表示应使用默认参数
])
def test_get_child_combined(self, adults_only_param, expected):
"""
结合哨兵值和kwargs,在一个参数化测试中覆盖默认参数。
"""
kwargs = {}
# 如果adults_only_param不是None,则将其加入kwargs字典
if adults_only_param is not None:
kwargs["adults_only"] = adults_only_param
# 调用get_user函数,并使用**kwargs解包关键字参数
actual = get_user(self.child.id, **kwargs)
self.assertEqual(expected, actual)
# 运行测试(可选)
if __name__ == '__main__':
unittest.main(argv=['first-arg-is-ignored'], exit=False)kwargs = {}
if adults_only_param is not None:
kwargs["adults_only"] = adults_only_param在每个测试用例执行时,我们首先初始化一个空的kwargs字典。然后,检查adults_only_param是否为None。如果不是None(即True或False),则将"adults_only": adults_only_param添加到kwargs中。
actual = get_user(self.child.id, **kwargs)
**kwargs语法用于将字典中的键值对解包为函数的关键字参数。
注意事项:
通过在参数化测试中巧妙地引入哨兵值并结合**kwargs动态构建参数,我们能够有效地将Python函数默认参数的测试合并到单个参数化测试用例中。这种方法不仅减少了测试代码的重复性,提高了测试套件的简洁性和可维护性,同时也确保了对默认参数行为的全面覆盖。在设计测试用例时,应优先考虑这种集成策略,以构建更高效、更易于管理的测试代码。
以上就是Python函数默认参数的参数化测试策略的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号