单元测试举例

以下是一个简单的 Python 函数，用于计算给定列表中所有元素的平均值：

def calculate_average(numbers):
    sum = 0
    count = 0
    for num in numbers:
        sum += num
        count += 1
    if count == 0:
        return 0
    else:
        return sum / count

现在，我们可以使用单元测试来测试这个函数是否按照预期进行运行。以下是一个示例测试用例：

import unittest

class TestCalculateAverage(unittest.TestCase):

    def test_calculate_average(self):
        numbers = [1, 2, 3, 4, 5]
        result = calculate_average(numbers)
        self.assertEqual(result, 3.0)

    def test_calculate_average_with_empty_list(self):
        numbers = []
        result = calculate_average(numbers)
        self.assertEqual(result, 0)

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

在这个示例代码中，我们使用 Python 的内置 unittest 模块编写了两个测试用例：

1. test_calculate_average：测试函数在给定列表中正确计算平均值的情况。

2. test_calculate_average_with_empty_list：测试函数在给定空列表时返回 0 的情况。

我们可以运行这个测试套件来验证函数是否按照预期进行运行。如果测试通过，我们可以相对自信地认为这个函数在不同的情况下都能够正确地计算平均值。如果测试失败，我们可以查看测试报告并分析错误，然后进行修复。