Как запустить код очистки только для определенных тестов с pytest?

Question

В pytest есть способ запустить код очистки только для определенной тестовой функции / метода. Я знаю, что мы можем сделать это для запуска каждой тестовой функции. Но здесь я хочу разместить некоторую логику очистки, специфичную для одной тестовой функции.

Я могу просто добавить код очистки в конце теста. Но если тест не пройден, очистка не будет выполнена.

Answer 1

Создайте прибор с вашим кодом очистки и внедрите его только в один тест, используя прибор в качестве аргумента для вашего теста или явно пометив тест с помощью декоратора pytest.mark.usefixtures.

import pytest

@pytest.fixture
def my_cleanup_fixture():
    # Startup code
    ...
    yield
    # Cleanup code
    ...

@pytest.mark.usefixtures('my_cleanup_fixture')
def test_with_special_cleanup():
    pass

my_cleanup_fixture имеет область действия function по умолчанию, поэтому код запуска и очистки будет выполняться для каждой введенной функции.