Reusable setup and teardown for pytest

test_setup_and_teardown.py

import pytest


def populate_tables(tables):
    pass


def truncate_tables(tables):
    pass


@pytest.fixture()
def database_fixture(request):
    tables = request.param.get('tables', {})
    populate_tables(tables)
    yield  # Teardown
    truncate_tables(tables.keys())


@pytest.mark.parametrize('database_fixture', [
    # Only one use case, since we are using values in fixture instead of
    # testing method itself
    (
        {
            'tables': {
                'user': {
                    'id': 1,
                    'name': 'Foobar'
                }
            }
        },
    )
], indirect=True)
def test_setup_and_teardown(database_fixture):
    pass