Merge pull request #26 from nlocascio/nl/fix_assignment_exception

Fix assignment exception

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pytest-alembic"
-version = "0.3.1"
+version = "0.3.2"
 description = "A pytest plugin for verifying alembic migrations."
 authors = [
     "Dan Cardin <ddcardin@gmail.com>",

src/pytest_alembic/tests.py

@@ -91,20 +91,20 @@ def test_up_down_consistency(alembic_runner):
 
     Individually upgrade to ensure that it's clear which revision caused the failure.
     """
-    try:
-        for revision in alembic_runner.history.revisions:
+    for revision in alembic_runner.history.revisions:
+        try:
             alembic_runner.migrate_up_to(revision)
-    except RuntimeError as e:
-        raise AlembicTestFailure(
-            "Failed to upgrade through each revision individually.",
-            context=[("Failing Revision", revision), ("Alembic Error", str(e))],
-        )
-
-    try:
-        for revision in reversed(alembic_runner.history.revisions):
+        except RuntimeError as e:
+            raise AlembicTestFailure(
+                "Failed to upgrade through each revision individually.",
+                context=[("Failing Revision", revision), ("Alembic Error", str(e))],
+            )
+
+    for revision in reversed(alembic_runner.history.revisions):
+        try:
             alembic_runner.migrate_down_to(revision)
-    except RuntimeError as e:
-        raise AlembicTestFailure(
-            "Failed to downgrade through each revision individually.",
-            context=[("Failing Revision", revision), ("Alembic Error", str(e))],
-        )
+        except RuntimeError as e:
+            raise AlembicTestFailure(
+                "Failed to downgrade through each revision individually.",
+                context=[("Failing Revision", revision), ("Alembic Error", str(e))],
+            )