python pytest如何編寫測(cè)試

小樊
81
2024-11-16 00:23:17

要使用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.pytest文件夾的目錄),然后運(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è)試用例。

0