Теория: Фреймворк Pytest

Когда тестов и файлов с тестами становится много, возникают новые вопросы:

• Как группировать тесты?
• Как запустить на выполнение все тесты из одной директории?
• Если тестов очень много и они долгие, можно ли запустить их параллельно?

Для решения этих вопросов используют специальные тестовые фреймворки. Они помогают организовать структуру тестов и дают много полезного, например, удобный вывод. С большинством из этих возможностей мы познакомимся далее по курсу.

В Python-мире популярен фреймворк Pytest, на котором мы сосредоточимся в этом уроке.

Что такое Pytest

Python — один из немногих языков, в стандартную библиотеку которого уже включен тестовый фреймворк (он называется unittest). Было бы логично рассмотреть его здесь, но он глубоко завязан на классы, которые вы можете не знать. Поэтому здесь мы рассматриваем Pytest, который все равно популярнее встроенного фреймворка.

Далее мы создадим проект с нуля и добавим в него тесты.

Настройка и запуск

Создайте где-нибудь на вашем компьютере директорию с названием hexlet_pytest. Зайдите в нее и выполните команды:

uv init # Создание базовой структуры
mkdir tests

Мы инициализируем новый проект и добавляем в него директорию для будущих тестов tests. Установите pytest с помощью uv в девелопмент зависимости с помощью команды uv add --dev pytest. Проверьте, что все работает правильно:

uv run pytest
# Ниже примерный вывод
platform linux -- Python 3.13.0, pytest-8.3.3, pluggy-1.5.0
rootdir: /private/tmp/hexlet_pytest
collected 0 items

======= no tests ran in 0.01s =======

Также добавьте в pyproject.toml блок. В строчках ниже мы задаем систему сборки пакета, а также указываем по какому пути взять файлы, что попадут в пакет. :

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.wheel]
packages = ["hexlet_pytest"]

Теперь добавим немного исходного кода. Создайте вложенную директорию hexlet_pytest, где будет храниться код проекта. Создайте в ней файл example.py.

В итоге, у вас должна получиться следующая структура:

.
├── hexlet_pytest
│   └── example.py
├── pyproject.toml
├── README.md
└── uv.lock

Внутри файла запишем код:

# Эта функция переворачивает переданную строку
def reverse(string):
    return string[::-1]

Pytest ожидает, что тесты находятся в директории tests, которую располагают в корне проекта. Внутри этого каталога можно создавать любую структуру — Pytest найдет все тесты, которые там лежат.

Именование файлов с тестами должно быть таким: test_<name>.py. Как правило, в этой записи <name> соответствует имени файла, который тестируется.

Напишем наш первый тест. Создайте файл tests/test_example.py со следующим содержимым:

from hexlet_pytest.example import reverse


def test_reverse():
    assert reverse("Hexlet") == "telxeH"


def test_reverse_for_empty_string():
    assert reverse("") == ""

Далее мы разберем структуру этого файла, а пока попробуем запустить тест на выполнение:

uv run pytest
platform linux -- Python 3.13.0, pytest-8.3.3, pluggy-1.5.0
rootdir: /private/tmp/hexlet-pytest
collected 2 items

tests/test_example.py ..       [100%]

# Один тест запускается так
# uv run pytest tests/test_example.py

Структура

Давайте еще раз посмотрим на файл с тестом:

from hexlet_pytest.example import reverse


def test_reverse():
    assert reverse("Hexlet") == "telxeH"


def test_reverse_for_empty_string():
    assert reverse("") == ""

Каждый тест внутри файла — это функция, начинающаяся с префикса test_. Тестовых функций можно делать любое количество, все уже зависит от сложности тестируемого кода. Внутри функций проверки выполняются с помощью инструкции assert.

Несмотря на схожесть со встроенной инструкцией в Python, инструкция assert в Pytest работает немного по-другому. Попробуйте внести ошибку в исходную функцию и запустите тесты заново:

def test_reverse():
>       assert reverse('Hexlet') == 'telxeH'
E       AssertionError: assert 'Hexlet' == 'telxeH'
E         - telxeH
E         + Hexlet

tests/test_example.py:4: AssertionError

Вывод Pytest содержит не только исходное утверждение, но и результат вызова функции:

• Плюсом помечается то, что пришло
• Минусом помечается то, что ожидалось

Благодаря такому выводу, найти и исправить ошибку становится значительно проще.

В работе Pytest есть одна особенность. Если понадобится распечатать что-то на экран, то Pytest ничего не выведет. По умолчанию он перехватывает и подавляет весь вывод. Отключить это поведение можно с помощью флага -s:

uv run pytest -s
# Теперь видно все, что выводится с помощью print