Как проверять код на Python: основные методы и подходы

Расширенные примеры проверки кода

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

1. Параметризованные тесты в pytest с фикстурами

import pytest

@pytest.fixture
def sample_data():
    return {'a': 10, 'b': 20}

@pytest.mark.parametrize('key, expected', [
    ('a', 10),
    ('b', 20),
    ('c', KeyError)
])
def test_fixture_values(sample_data, key, expected):
    if key == 'c':
        with pytest.raises(KeyError):
            _ = sample_data[key]
    else:
        assert sample_data[key] == expected

$ pytest test_param.py -v
============================= test session starts ==============================
collected 3 items

test_param.py::test_fixture_values[a-10] PASSED
test_param.py::test_fixture_values[b-20] PASSED
test_param.py::test_fixture_values[c-KeyError] PASSED

============================== 3 passed in 0.02s ===============================

2. Тестирование с unittest и mock

import unittest
from unittest.mock import patch
from math import sqrt

def fetch_value(calculator, x):
    return calculator.sqrt(x)

class TestWithMock(unittest.TestCase):
    @patch('math.sqrt')
    def test_fetch(self, mock_sqrt):
        mock_sqrt.return_value = 25
        result = fetch_value(sqrt, 100)  # передаем реальную функцию, но она заменена mock
        self.assertEqual(result, 25)
        mock_sqrt.assert_called_once_with(100)

if __name__ == '__main__':
    unittest.main()

$ python test_mock.py
.
----------------------------------------------------------------------
Ran 1 test in 0.001s

OK

3. Доктест с обработкой исключений

def safe_divide(a, b):
    """
    Делит a на b, возвращает None при ошибке.

    >>> safe_divide(10, 2)
    5.0
    >>> safe_divide(10, 0) is None
    True
    >>> safe_divide('a', 2)  # doctest: +IGNORE_EXCEPTION_DETAIL
    Traceback (most recent call last):
    TypeError: unsupported operand type(s) for /: 'str' and 'int'
    """
    try:
        return a / b
    except ZeroDivisionError:
        return None

if __name__ == '__main__':
    import doctest
    doctest.testmod(verbose=True)

$ python doctest_adv.py
Trying:
    safe_divide(10, 2)
Expecting:
    5.0
ok
Trying:
    safe_divide(10, 0) is None
Expecting:
    True
ok
Trying:
    safe_divide('a', 2)
Expecting:
    Traceback (most recent call last):
    TypeError: unsupported operand type(s) for /: 'str' and 'int'
ok
2 items had no tests:
    __main__
    __main__.safe_divide
1 items passed all tests:
   3 tests in __main__.safe_divide
3 tests in 3 items.
3 passed and 0 failed.
Test passed.

4. Статический анализ с mypy для класса

from typing import List, Optional

class Person:
    def __init__(self, name: str, age: Optional[int] = None) -> None:
        self.name = name
        self.age = age

    def greet(self) -> str:
        if self.age:
            return f'Привет, меня зовут {self.name}, мне {self.age} лет'
        return f'Привет, я {self.name}'

def introduce(people: List[Person]) -> None:
    for person in people:
        print(person.greet())

# Ошибка: передаётся число вместо списка
introduce(123)

$ mypy class_example.py
class_example.py:18: error: Argument 1 to "introduce" has incompatible type "int"; expected "List[Person]"
Found 1 error in 1 file (checked 1 source file)

5. Пошаговая отладка с pdb и условными брейкпоинтами

def complex_loop(n):
    total = 0
    for i in range(n):
        total += i
        if i == 3:
            import pdb
            pdb.set_trace()  # остановка только при i == 3
        total *= 1
    return total

complex_loop(10)

$ python pdb_conditional.py
> /Users/user/pdb_conditional.py(7)complex_loop()
-> total *= 1
(Pdb) p i
3
(Pdb) p total
6
(Pdb) c

6. Логирование с ротацией файлов и разными форматтерами

import logging
from logging.handlers import RotatingFileHandler

logger = logging.getLogger('myapp')
logger.setLevel(logging.DEBUG)

# Консольный вывод
console = logging.StreamHandler()
console.setLevel(logging.INFO)
formatter = logging.Formatter('%(name)s - %(levelname)s - %(message)s')
console.setFormatter(formatter)
logger.addHandler(console)

# Файл с ротацией
file_handler = RotatingFileHandler('app.log', maxBytes=100, backupCount=2)
file_handler.setLevel(logging.DEBUG)
file_formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
file_handler.setFormatter(file_formatter)
logger.addHandler(file_handler)

logger.debug('Отладочное сообщение')
logger.info('Информация')
logger.warning('Предупреждение')
logger.error('Ошибка')

Консоль:
myapp - INFO - Информация
myapp - WARNING - Предупреждение
myapp - ERROR - Ошибка

Файл app.log:
2025-03-27 20:15:30,123 - myapp - DEBUG - Отладочное сообщение
2025-03-27 20:15:30,123 - myapp - INFO - Информация
2025-03-27 20:15:30,123 - myapp - WARNING - Предупреждение
2025-03-27 20:15:30,123 - myapp - ERROR - Ошибка