fix: update _patch_bases fixture to yield and clean up registries after tests

Author: cofin

tests/integration/conftest.py

@@ -89,7 +89,7 @@ def cleanup() -> None:
 
 
 @pytest.fixture(autouse=True)
-def _patch_bases(monkeypatch: MonkeyPatch) -> None:  # pyright: ignore[reportUnusedFunction]
+def _patch_bases(monkeypatch: MonkeyPatch) -> Generator[None, None, None]:  # pyright: ignore[reportUnusedFunction]
     """Ensure new registry state for every test.
 
     This prevents errors such as "Table '...' is already defined for
@@ -155,6 +155,24 @@ class NewBigIntAuditBase(
     monkeypatch.setattr(base, "BigIntBase", NewBigIntBase)
     monkeypatch.setattr(base, "BigIntAuditBase", NewBigIntAuditBase)
 
+    yield
+
+    # Clean up registries to prevent test pollution
+    for base_class in [
+        NewUUIDBase,
+        NewUUIDAuditBase,
+        NewUUIDv6Base,
+        NewUUIDv6AuditBase,
+        NewUUIDv7Base,
+        NewUUIDv7AuditBase,
+        NewNanoIDBase,
+        NewNanoIDAuditBase,
+        NewBigIntBase,
+        NewBigIntAuditBase,
+    ]:
+        base_class.registry.dispose()
+        base_class.metadata.clear()
+
 
 @pytest.fixture(scope="session")
 def duckdb_engine(