在Python中,进行单元测试通常使用内置的unittest
模块。以下是一个简单的示例,展示了如何使用unittest
模块进行单元测试:
- 首先,创建一个名为
calculator.py
的文件,其中包含要测试的函数。例如,我们创建一个简单的计算器函数,用于执行加法和乘法操作:
# calculator.py def add(a, b): return a + b def multiply(a, b): return a * b
- 接下来,创建一个名为
test_calculator.py
的文件,其中包含针对calculator.py
中函数的单元测试:
# test_calculator.py import unittest from calculator import add, multiply class TestCalculator(unittest.TestCase): def test_addition(self): self.assertEqual(add(1, 2), 3) self.assertEqual(add(-1, 1), 0) self.assertEqual(add(0, 0), 0) def test_multiplication(self): self.assertEqual(multiply(1, 2), 2) self.assertEqual(multiply(-1, 1), -1) self.assertEqual(multiply(0, 0), 0) if __name__ == '__main__': unittest.main()
在这个例子中,我们创建了一个名为TestCalculator
的测试类,该类继承自unittest.TestCase
。我们编写了两个测试方法test_addition
和test_multiplication
,分别用于测试add
和multiply
函数。在每个测试方法中,我们使用self.assertEqual
来检查预期结果与实际结果是否相等。
- 最后,在命令行中运行
test_calculator.py
文件以执行单元测试:
python test_calculator.py
如果所有测试都通过,你将看到以下输出:
.... ---------------------------------------------------------------------- Ran 4 tests in 0.001s OK
这表明我们的calculator.py
文件中的函数已成功通过了单元测试。