Snippets for Pytest fixtures.

pytest-sqlalchemy.py

"""Declare fixtures for SQLAlchemy (session and engine)."""

# IMPORTS

import sqlalchemy as sql
from sqlalchemy import orm

from project import models  # TODO

# FIXTURES

@pytest.fixture(scope="function")
def engine() -> sql.engine.Engine:
    """Create an SQLAlchemy engine for testing purposes."""
    engine = sql.create_engine("sqlite:///:memory:")
    return engine


@pytest.fixture(scope="function")
def session(engine: sql.engine.Engine) -> T.Iterator[orm.Session]:
    """Create an SQLAlchemy session for testing purposes."""
    models.Base.metadata.create_all(engine)
    with orm.Session(engine) as session_:
        yield session_
        session_.commit()