Merge pull request #125 from ecmwf-projects/fix-alembic-tests

fix alembic tests
ecmwf-projects · Aug 1, 2024 · d44202c · d44202c
2 parents dc4d0c5 + f0bc98d
commit d44202c
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/test_90_entry_points.py b/tests/test_90_entry_points.py
@@ -88,9 +88,12 @@ def test_init_db(postgresql: Connection[str], mocker) -> None:
     )
     assert set(conn.execute(query).scalars()) == set(
         database.BaseModel.metadata.tables
-    ).union({"alembic_version_cacholote"}).union(
-        set(cacholote.database.Base.metadata.tables)
-    )
+    ).union(
+        {
+            "alembic_version",
+            "alembic_version_cacholote",
+        }
+    ).union(set(cacholote.database.Base.metadata.tables))
     conn.close()