Как протестировать код на Python с pytest

Тестирование кода — важная часть разработки программного обеспечения, которая помогает обеспечить работоспособность и надежность вашего кода. В этой статье мы рассмотрим, как использовать инструмент для тестирования под названием pytest для проверки вашего кода на Python.

Установка pytest

Для начала, вам нужно установить pytest. Это можно сделать с помощью следующей команды:

pip install pytest

Написание тестов

Теперь, когда у вас установлен pytest, давайте напишем простой тест. Создайте файл с именем test_example.py и добавьте следующий код:

def test_addition():
    assert 1 + 1 == 2

Здесь мы создали функцию test_addition, которая проверяет, верно ли выполняется сложение двух чисел. Обратите внимание на префикс test_ в имени функции — это необходимо для того, чтобы pytest смог автоматически распознать и запустить этот тест.

Запуск тестов

Чтобы запустить тесты с помощью pytest, просто выполните команду pytest в терминале. Вам должны быть показаны результаты тестирования, включая количество успешных и проваленных тестов.

pytest

😉 Если все сделано правильно, вы увидите сообщение о том, что ваш тест успешно пройден.

Расширенное использование

pytest предлагает множество возможностей для написания и организации тестов. Например, вы можете использовать параметризацию для выполнения одного и того же теста с разными входными данными:

import pytest

@pytest.mark.parametrize("input_a, input_b, expected_output", [
    (1, 1, 2),
    (2, 2, 4),
    (3, 5, 8),
])
def test_addition(input_a, input_b, expected_output):
    assert input_a + input_b == expected_output

В этом примере мы используем декоратор @pytest.mark.parametrize для указания различных наборов входных данных и ожидаемых результатов. Тест будет выполнен для каждого набора данных, что позволяет легко проверить различные случаи.

Заключение

В этой статье мы кратко рассмотрели, как использовать pytest для тестирования вашего кода на Python. Это мощный инструмент, который может облегчить вашу работу и улучшить качество вашего кода. Не стесняйтесь изучать дополнительные возможности pytest и интегрировать его в свои проекты. Удачного кодирования!