Python：unittest 库高级用法举例和应用详解

import unittest  # 导入unittest模块

# 定义一个简单的数学函数库
class MathLib:
    def add(a, b):
        return a + b

    def subtract(a, b):
        return a - b

# 编写测试类，继承unittest.TestCase
class TestMathLib(unittest.TestCase):

    def test_add(self):
        # 测试加法函数
        self.assertEqual(MathLib.add(1, 2), 3)  # 测试1+2是否等于3

    def test_subtract(self):
        # 测试减法函数
        self.assertEqual(MathLib.subtract(3, 2), 1)  # 测试3-2是否等于1

if __name__ == '__main__':
    unittest.main()  # 运行所有的测试用例

import unittest  # 导入unittest模块

# 定义一个函数，除以零时抛出异常
def divide(a, b):
    if b == 0:
        raise ValueError("Cannot divide by zero")
    return a / b

# 编写测试类
class TestDivision(unittest.TestCase):

    def test_divide_by_zero(self):
        # 检查是否抛出ValueError异常
        with self.assertRaises(ValueError):
            divide(1, 0)  # 1除以0，应该抛出ValueError

if __name__ == '__main__':
    unittest.main()  # 运行所有的测试用例

import unittest  # 导入unittest模块

# 定义一个函数，统计列表中元素出现的频率
def count_elements(data):
    counter = {}
    for item in data:
        if item in counter:
            counter[item] += 1
        else:
            counter[item] = 1
    return counter

# 编写测试类
class TestCountElements(unittest.TestCase):

    def test_count_elements(self):
        # 测试列表中元素的计数
        data = ['apple', 'banana', 'apple', 'orange', 'banana', 'apple']
        result = count_elements(data)
        expected_result = {'apple': 3, 'banana': 2, 'orange': 1}
        self.assertEqual(result, expected_result)  # 比较结果是否相同

if __name__ == '__main__':
    unittest.main()  # 运行所有的测试用例