在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
文件中的函数已成功通过了单元测试。