Python中如何为多返回值方法编写Unittest测试用例

当然可以啦！完全不用拆分原方法，Python的unittest框架直接就能处理这类场景——因为Python里函数返回多个值本质上是返回一个元组，咱们直接针对这个元组做断言就好，两种方式都很顺手：

方式1：解包后分别断言每个返回值 #

这种方式适合你需要单独验证每个返回结果的场景，可读性更强。先看一个示例：

待测试的方法 #

def calculate_stats(numbers):
    """计算一组数字的总和和平均值"""
    total = sum(numbers)
    average = total / len(numbers) if numbers else 0
    return total, average  # 本质返回的是元组 (total, average)

对应的测试用例 #

import unittest

class TestCalculateStats(unittest.TestCase):
    def test_normal_dataset(self):
        # 调用方法并解包返回值
        actual_total, actual_avg = calculate_stats([1, 2, 3, 4])
        # 分别断言每个结果
        self.assertEqual(actual_total, 10)
        self.assertEqual(actual_avg, 2.5)
        
    def test_single_number(self):
        total, avg = calculate_stats([7])
        self.assertEqual(total, 7)
        self.assertEqual(avg, 7.0)

方式2：直接断言整个返回的元组 #

如果只需要验证整体结果是否符合预期，直接把返回值和预期的元组做断言更简洁：

class TestCalculateStats(unittest.TestCase):
    def test_empty_dataset(self):
        # 直接对比返回的元组和预期元组
        self.assertEqual(calculate_stats([]), (0, 0))

其他情况拓展 #

如果你的方法返回的是列表、字典这类复杂的多值结构（比如return [1,2], {"name": "test"}），同样可以用类似思路：要么解包后分别断言每个结构，要么直接断言整个返回的元组是否与预期一致。

内容的提问来源于stack exchange，提问作者harry77