python pytest如何編寫測試

要使用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中的其他函數編寫更多的測試用例。