Comment utiliser pytest pour les tests unitaires Python

Comment utiliser pytest pour les tests unitaires Python

Introduction :
Les tests unitaires sont une tâche indispensable dans le développement de logiciels. Ils peuvent nous aider à garantir la qualité et la stabilité du code. En Python, pytest est un framework de tests unitaires puissant et facile à utiliser. Cet article explique comment utiliser pytest pour les tests unitaires Python.

1. Installez pytest
Tout d'abord, nous devons installer pytest. L'installation peut être facilement complétée à l'aide de la commande pip :

pip install pytest

2. Écrire des cas de test
Ensuite, nous devons écrire des cas de test. Un scénario de test est un bloc de code qui vérifie la fonction ou la fonctionnalité du code testé.

Exemple :
Supposons que nous voulions tester un module de calculatrice simple, qui contient quatre opérations : addition, soustraction, multiplication et division. Nous pouvons créer un fichier nommé test_calculator.py dans le projet pour stocker le code du scénario de test. Voici le code d'un exemple de cas de test : 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

Dans le code ci-dessus, nous utilisons l'instruction assert fournie par le framework pytest pour affirmer si les résultats du test sont comme prévu.

3. Exécuter des cas de test

Après avoir terminé la rédaction des cas de test, nous pouvons utiliser pytest pour exécuter ces cas de test. 🎜🎜Entrez le répertoire du projet sur la ligne de commande et entrez la commande suivante pour tester : 🎜rrreee🎜pytest trouvera automatiquement les cas de test dans le projet et les exécutera. 🎜🎜Exemple de résultats en cours d'exécution : 🎜rrreee🎜Les résultats ci-dessus montrent que les 4 cas de test que nous avons écrits ont réussi le test. 🎜🎜4. Utilisation avancée🎜

1. Tests paramétrés🎜Parfois, afin de vérifier le fonctionnement d'une fonction sous différents paramètres d'entrée, nous devons écrire plusieurs cas de test. pytest fournit la fonction de tests paramétrés, ce qui peut réduire la charge de travail liée à l'écriture de code répété.

🎜Exemple : 🎜rrreee🎜Dans le code ci-dessus, nous utilisons le décorateur @pytest.mark.parametrize pour paramétrer les cas de test. Chaque ensemble de paramètres est automatiquement transmis à la fonction testée et des assertions sont faites. 🎜

2. Définir les pré- et post-opérations 🎜Pendant l'exécution des scénarios de test, nous devons parfois effectuer un travail préparatoire avant l'exécution du scénario de test, et en effectuer quelques-uns après l'exécution du scénario de test. Clean en haut. pytest fournit des décorateurs tels que setup et teardown pour implémenter ces opérations.

🎜Exemple : 🎜rrreee🎜Dans l'exemple ci-dessus, nous utilisons le décorateur setup_and_teardown pour définir une fonction avec des pré- et post-opérations. Utilisez le mot-clé yield pour insérer le scénario de test à l'emplacement où la fonction est définie. 🎜🎜Conclusion : 🎜Cet article explique comment utiliser pytest pour les tests unitaires Python. Vous pouvez utiliser pytest pour effectuer facilement des tests de code en installant pytest, en écrivant des scénarios de test, en exécutant des scénarios de test et en utilisant une utilisation avancée. J'espère que cet article pourra être utile pour votre travail de tests unitaires dans le développement de logiciels. 🎜

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!