如何在Pytest测试后彻底删除SQLAlchemy动态创建的模型类？

动态创建的SQLAlchemy模型会被元数据（MetaData）和ORM注册表（Registry）保留，导致测试间出现副作用（比如重复创建同名模型时抛出异常）。可以通过以下步骤在每次测试后彻底清理这些动态模型：

1. 清理元数据中的表对象 #

SQLAlchemy的元数据会跟踪所有关联的表，测试结束后需要从基类的metadata中移除对应表：

# obj是你动态创建的模型类
BaseModelDerivedFromDeclarativeBase.metadata.remove(obj.__table__)

如果测试中已经将表创建到数据库，还需要删除数据库中的表：

obj.__table__.drop(BaseModelDerivedFromDeclarativeBase.engine)

2. 移除ORM注册表中的模型类 #

SQLAlchemy 2.0+的DeclarativeBase基于Registry管理模型类，需要从注册表的私有_class_registry中删除模型：

registry = BaseModelDerivedFromDeclarativeBase.registry
if obj.__name__ in registry._class_registry:
    del registry._class_registry[obj.__name__]

如果是SQLAlchemy 1.x版本，操作基类的_decl_class_registry即可：

if obj.__name__ in BaseModelDerivedFromDeclarativeBase._decl_class_registry:
    del BaseModelDerivedFromDeclarativeBase._decl_class_registry[obj.__name__]

3. 用Pytest自动清理fixture简化操作 #

创建一个自动生效的fixture，统一处理测试前后的模型清理，避免重复代码：

import pytest
from sqlalchemy import Column, Integer, String
from your_module import BaseModelDerivedFromDeclarativeBase

@pytest.fixture(autouse=True)
def cleanup_dynamic_models():
    # 存储当前测试创建的动态模型
    created_models = []
    yield created_models
    # 测试结束后执行清理
    for model in created_models:
        # 清理元数据和数据库表
        if model.__table__ in BaseModelDerivedFromDeclarativeBase.metadata.tables.values():
            BaseModelDerivedFromDeclarativeBase.metadata.remove(model.__table__)
        model.__table__.drop(BaseModelDerivedFromDeclarativeBase.engine)
        # 清理注册表
        registry = BaseModelDerivedFromDeclarativeBase.registry
        if model.__name__ in registry._class_registry:
            del registry._class_registry[model.__name__]

在测试中创建动态模型时，只需将模型添加到fixture的列表中：

def test_dynamic_model(cleanup_dynamic_models):
    obj_args = {
        "__tablename__": "t_test",
        "__table_args__": {"extend_existing": True},
        "id": Column(Integer, primary_key=True),
        "name": Column(String)
    }
    test_model = type("TestModel", (BaseModelDerivedFromDeclarativeBase,), obj_args)
    cleanup_dynamic_models.append(test_model)
    
    # 执行测试逻辑...

额外注意事项 #

• 优先使用独立的测试数据库（比如SQLite内存库、临时PostgreSQL库），即使清理不彻底，也不会影响其他测试
• 尽量给每个测试的动态模型生成唯一名称（比如拼接测试函数名、随机字符串），从根源避免同名冲突

内容的提问来源于stack exchange，提问作者Roman Purgstaller