Cara menggunakan pytest untuk ujian unit Python

Cara menggunakan pytest untuk ujian unit Python

Pengenalan:
Ujian unit ialah tugas yang sangat diperlukan dalam pembangunan perisian. Ia boleh membantu kami memastikan kualiti dan kestabilan kod. Dalam Python, pytest ialah rangka kerja ujian unit yang berkuasa dan mudah digunakan. Artikel ini akan memperkenalkan cara menggunakan pytest untuk ujian unit Python.

1. Pasang pytest
Pertama, kita perlu memasang pytest. Pemasangan boleh diselesaikan dengan mudah menggunakan arahan pip:

pip install pytest

2. Tulis kes ujian
Seterusnya, kita perlu menulis kes ujian. Kes ujian ialah blok kod yang mengesahkan fungsi atau kefungsian kod yang sedang diuji.

Contoh:
Andaikan kita ingin menguji modul kalkulator mudah, yang mengandungi empat operasi: tambah, tolak, darab dan bahagi. Kami boleh mencipta fail bernama test_calculator.py dalam projek untuk menyimpan kod kes ujian. Berikut ialah kod contoh kes ujian: test_calculator.py的文件，用于存放测试用例的代码。以下是一个示例测试用例的代码：

# test_calculator.py

from calculator import add, subtract, multiply, divide

def test_add():
    assert add(2, 3) == 5

def test_subtract():
    assert subtract(5, 3) == 2

def test_multiply():
    assert multiply(2, 3) == 6

def test_divide():
    assert divide(6, 3) == 2

在上述代码中，我们使用了pytest框架提供的assert语句来断言测试结果是否符合预期。

三、运行测试用例
完成测试用例的编写后，我们可以使用pytest来运行这些测试用例。

在命令行中进入项目目录，输入以下命令进行测试：

pytest

pytest会自动查找项目中的测试用例并运行。

运行结果示例：

============================= test session starts ==============================
platform darwin -- Python 3.9.0, pytest-6.2.2, py-1.10.0, pluggy-0.13.1
rootdir: /path/to/project
collected 4 items                                                              

test_calculator.py ....                                                  [100%]

============================== 4 passed in 0.02s ===============================

上述结果说明我们编写的4个测试用例全部通过了测试。

四、高级用法

1. 参数化测试
   有时候，为了验证一个函数在不同的输入参数下的工作情况，我们需要编写多个测试用例。pytest提供了参数化测试的功能，可以减少编写重复代码的工作量。

示例：

import pytest

@pytest.mark.parametrize(
    "a, b, expected",
    [
        (2, 3, 5),
        (-1, 1, 0),
        (0, 0, 0),
    ],
)
def test_add(a, b, expected):
    assert add(a, b) == expected

上述代码中，我们使用@pytest.mark.parametrize装饰器来参数化测试用例。每组参数都会被自动传递给被测试的函数，并进行断言判断。

2. 设置前置和后置操作
   在测试用例的执行过程中，有时候我们需要在测试用例运行之前执行一些准备工作，以及在测试用例运行之后执行一些清理工作。pytest提供了setup和teardown等装饰器来实现这些操作。

示例：

@pytest.fixture(autouse=True)
def setup_and_teardown():
    # 在测试用例运行之前的准备工作
    print("准备工作") 

    yield

    # 在测试用例运行之后的清理工作
    print("清理工作")

在上述示例中，我们使用setup_and_teardown装饰器来定义一个前置和后置操作的函数。使用yieldrrreee

Dalam kod di atas, kami menggunakan pernyataan assert yang disediakan oleh rangka kerja pytest untuk menegaskan sama ada keputusan ujian adalah seperti yang dijangkakan.

3. Jalankan kes ujian

Selepas selesai menulis kes ujian, kita boleh menggunakan pytest untuk menjalankan kes ujian ini. 🎜🎜Masukkan direktori projek pada baris arahan dan masukkan arahan berikut untuk menguji: 🎜rrreee🎜pytest secara automatik akan mencari kes ujian dalam projek dan menjalankannya. 🎜🎜Contoh keputusan berjalan: 🎜rrreee🎜Keputusan di atas menunjukkan bahawa kesemua 4 kes ujian yang kami tulis lulus ujian. 🎜🎜4. Penggunaan lanjutan🎜

1. Ujian berparameter🎜Kadangkala, untuk mengesahkan fungsi fungsi di bawah parameter input yang berbeza, kita perlu menulis beberapa kes ujian. pytest menyediakan fungsi ujian parameter, yang boleh mengurangkan beban kerja menulis kod berulang.

🎜Contoh: 🎜rrreee🎜Dalam kod di atas, kami menggunakan penghias @pytest.mark.parametrize untuk membuat parameter kes ujian. Setiap set parameter dihantar secara automatik ke fungsi di bawah ujian dan penegasan dibuat. 🎜

2. Tetapkan pra dan pasca operasi 🎜Semasa pelaksanaan kes ujian, kadangkala kita perlu melakukan beberapa kerja persediaan sebelum kes ujian dijalankan dan melakukan beberapa kerja selepas kes ujian berjalan Bersih naik. pytest menyediakan penghias seperti setup dan teardown untuk melaksanakan operasi ini.

🎜Contoh: 🎜rrreee🎜Dalam contoh di atas, kami menggunakan penghias setup_and_teardown untuk mentakrifkan fungsi dengan pra dan pasca operasi. Gunakan kata kunci yield untuk memasukkan kes ujian ke lokasi di mana fungsi ditakrifkan. 🎜🎜Kesimpulan: 🎜Artikel ini menerangkan cara menggunakan pytest untuk ujian unit Python. Anda boleh menggunakan pytest untuk menjalankan ujian kod dengan mudah dengan memasang pytest, menulis kes ujian, menjalankan kes ujian dan menggunakan penggunaan lanjutan. Saya harap artikel ini boleh membantu kerja ujian unit anda dalam pembangunan perisian. 🎜

Atas ialah kandungan terperinci Cara menggunakan pytest untuk ujian unit Python. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!