Как параметризовать тест в Pytest?

В PyTest параметризованные тесты реализуются с помощью декоратора @pytest.mark.parametrize. Он позволяет запускать один и тот же тест с разными наборами входных данных.

Пример:

import pytest

@pytest.mark.parametrize("input,expected", [
    (2, 4),
    (3, 9),
    (4, 16),
])
def test_square(input, expected):
    assert input * input == expected

В этом примере тест test_square будет выполнен три раза с разными значениями input и expected. Это удобно для проверки функции на разных данных без дублирования кода.