在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
文件中的函數已成功通過了單元測試。