在Python Unittest中利用参数化测试高效处理默认参数

def get_user(user_id, adults_only=False):
    """
    根据用户ID获取用户信息。
    :param user_id: 用户唯一标识符。
    :param adults_only: 布尔值，如果为True则只返回成人信息，默认为False。
    :return: 模拟的用户信息。
    """
    # 实际实现可能涉及数据库查询等
    if user_id == 1:
        return "adult_info" if adults_only else "child_info"
    return "unknown_user"

import unittest
from parameterized import parameterized

# 假设这是我们要测试的函数（为示例方便，此处提供一个简单实现）
def get_user(user_id, adults_only=False):
    if user_id == 1:
        return "adult_info" if adults_only else "child_info"
    return "unknown_user"

class TestGetUserTraditional(unittest.TestCase):
    class MockChild:
        def __init__(self, id):
            self.id = id

    def setUp(self):
        self.child = self.MockChild(1)

    @parameterized.expand([
        (True, "adult_info"),
        (False, "child_info")
    ])
    def test_get_user_with_adults_only(self, adults_only_value, expected):
        actual = get_user(self.child.id, adults_only=adults_only_value)
        self.assertEqual(expected, actual)

    # 为了测试默认参数，需要单独编写一个测试
    def test_get_user_default_adults_only(self):
        actual = get_user(self.child.id) # 调用时省略 adults_only
        self.assertEqual("child_info", actual)

# if __name__ == '__main__':
#     unittest.main()

import unittest
from parameterized import parameterized

# 假设这是我们要测试的函数
def get_user(user_id, adults_only=False):
    if user_id == 1:
        return "adult_info" if adults_only else "child_info"
    return "unknown_user"

class TestGetUserOptimized(unittest.TestCase):
    class MockChild:
        def __init__(self, id):
            self.id = id

    def setUp(self):
        self.child = self.MockChild(1)

    @parameterized.expand([
        (True, "adult_info"),   # adults_only = True
        (False, "child_info"),  # adults_only = False
        (None, "child_info")    # 哨兵值None，表示使用默认参数
    ])
    def test_get_user_combined_adults_only(self, adults_only_param, expected):
        # 根据 adults_only_param 是否为 None 来动态构建 kwargs
        kwargs = {"adults_only": adults_only_param} if adults_only_param is not None else {}

        # 调用被测函数，使用 **kwargs 动态传递参数
        actual = get_user(self.child.id, **kwargs)
        self.assertEqual(expected, actual)

if __name__ == '__main__':
    unittest.main()