要使用Python的pytest編寫測試,首先確保已經安裝了pytest庫。如果尚未安裝,可以使用以下命令進行安裝:
pip install pytest
接下來,創建一個簡單的Python項目結構,例如:
my_project/
├── main.py
└── test/
└── test_main.py
在main.py
中,編寫一個簡單的函數,我們將在測試文件中對其進行測試:
# main.py
def add(a, b):
return a + b
現在,在test/test_main.py
文件中編寫測試用例。使用pytest
模塊作為測試文件的名稱前綴,并使用test_
作為函數名的前綴。在這個例子中,我們將編寫一個測試函數test_add
來測試main.py
中的add
函數:
# test/test_main.py
import pytest
from main import add
def test_add():
assert add(1, 2) == 3
assert add(-1, 1) == 0
assert add(0, 0) == 0
在這個測試函數中,我們使用了assert
語句來檢查add
函數的輸出是否與預期相符。如果輸出不符合預期,測試將失敗。
要運行測試,打開終端,導航到項目根目錄(包含main.py
和test
文件夾的目錄),然后運行以下命令:
pytest
pytest將自動發現并運行所有測試用例。如果所有測試都通過,你將看到類似以下的輸出:
=========================== test session starts ============================
...
collected 1 item
test/test_main.py . [100%]
============================ 1 passed in 0.03s =============================
這表明你已經成功地使用pytest編寫并運行了一個簡單的測試用例。你可以繼續為main.py
中的其他函數編寫更多的測試用例。