Python 测试用例

在Python中编写测试用例通常使用unittest模块，这是Python标准库的一部分，专门用于编写和运行测试。下面是一个简单的测试用例的例子，展示了如何使用unittest模块来测试一个函数。

假设我们有一个简单的函数，用于计算两个数的和：

def add(a, b):
    return a + b

我们可以编写以下测试用例来验证这个函数：

import unittest

class TestAddFunction(unittest.TestCase):

    def test_add_integers(self):
        self.assertEqual(add(1, 2), 3)

    def test_add_floats(self):
        self.assertAlmostEqual(add(0.1, 0.2), 0.3, places=1)

    def test_add_negative_numbers(self):
        self.assertEqual(add(-1, -1), -2)

# 运行测试
if __name__ == '__main__':
    unittest.main()

在这个例子中，我们定义了一个测试类TestAddFunction，它继承自unittest.TestCase。在这个类中，我们定义了三个测试方法：

• test_add_integers：测试两个整数相加的情况。
• test_add_floats：测试两个浮点数相加的情况，并使用assertAlmostEqual来考虑到浮点数的精度问题。
• test_add_negative_numbers：测试两个负数相加的情况。

unittest模块提供了多种断言方法，例如：

• assertEqual(a, b)：如果a和b相等，则测试通过。
• assertTrue(x)：如果x为True，则测试通过。
• assertFalse(x)：如果x为False，则测试通过。
• assertRaises(exception, callable, *args, **kwargs)：测试调用callable函数时是否会引发exception异常。
• assertAlmostEqual(a, b, places=7, msg=None, delta=None)：测试两个浮点数是否足够接近。

最后，通过if __name__ == '__main__':块，我们调用unittest.main()来运行测试用例。

这只是一个简单的例子，实际的测试用例可能包含更多的测试方法和复杂的测试逻辑。使用unittest模块，你可以构建强大的测试套件来确保你的代码按预期工作。