如何配置Postgres以使用pytest

# 安装PostgreSQL
sudo apt-get update
sudo apt-get install postgresql postgresql-contrib

# 安装Pytest
pip install pytest

sudo -u postgres psql

CREATE DATABASE mydatabase;
CREATE USER myuser WITH ENCRYPTED PASSWORD 'mypassword';
GRANT ALL PRIVILEGES ON DATABASE mydatabase TO myuser;
\q

pip install pytest-postgresql

[pytest]
addopts = --postgresql-host=localhost --postgresql-port=5432 --postgresql-database=mydatabase --postgresql-user=myuser --postgresql-password=mypassword

import pytest
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String)

@pytest.fixture(scope='module')
def db():
    engine = create_engine('postgresql://myuser:mypassword@localhost:5432/mydatabase')
    Base.metadata.create_all(engine)
    Session = sessionmaker(bind=engine)
    yield Session()
    Base.metadata.drop_all(engine)

def test_create_user(db):
    user = User(name='John Doe')
    db.add(user)
    db.commit()
    assert db.query(User).filter_by(name='John Doe').first() is not None