要使用Python的pytest編寫測(cè)試,首先確保已經(jīng)安裝了pytest庫(kù)。如果尚未安裝,可以使用以下命令進(jìn)行安裝:
pip install pytest
接下來(lái),創(chuàng)建一個(gè)簡(jiǎn)單的Python項(xiàng)目結(jié)構(gòu),例如:
my_project/
├── main.py
└── test/
└── test_main.py
在main.py
中,編寫一個(gè)簡(jiǎn)單的函數(shù),我們將在測(cè)試文件中對(duì)其進(jìn)行測(cè)試:
# main.py
def add(a, b):
return a + b
現(xiàn)在,在test/test_main.py
文件中編寫測(cè)試用例。使用pytest
模塊作為測(cè)試文件的名稱前綴,并使用test_
作為函數(shù)名的前綴。在這個(gè)例子中,我們將編寫一個(gè)測(cè)試函數(shù)test_add
來(lái)測(cè)試main.py
中的add
函數(shù):
# 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
在這個(gè)測(cè)試函數(shù)中,我們使用了assert
語(yǔ)句來(lái)檢查add
函數(shù)的輸出是否與預(yù)期相符。如果輸出不符合預(yù)期,測(cè)試將失敗。
要運(yùn)行測(cè)試,打開終端,導(dǎo)航到項(xiàng)目根目錄(包含main.py
和test
文件夾的目錄),然后運(yùn)行以下命令:
pytest
pytest將自動(dòng)發(fā)現(xiàn)并運(yùn)行所有測(cè)試用例。如果所有測(cè)試都通過(guò),你將看到類似以下的輸出:
=========================== test session starts ============================
...
collected 1 item
test/test_main.py . [100%]
============================ 1 passed in 0.03s =============================
這表明你已經(jīng)成功地使用pytest編寫并運(yùn)行了一個(gè)簡(jiǎn)單的測(cè)試用例。你可以繼續(xù)為main.py
中的其他函數(shù)編寫更多的測(cè)試用例。