Skip to content

Commit 5a17e3f

Browse files
committed
Fix(migration): mypy
1 parent 39e251a commit 5a17e3f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

migrations/versions/40_promo_above_2000.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,18 +84,18 @@ def test_upgrade(
8484
assert (
8585
alembic_connection.execute(
8686
sa.text(f"SELECT promo FROM core_user WHERE user_id = '{user_id_23}'"),
87-
).first()[0]
87+
).all()[0][0]
8888
== 2023
8989
)
9090
assert (
9191
alembic_connection.execute(
9292
sa.text(f"SELECT promo FROM core_user WHERE user_id = '{user_id_2023}'"),
93-
).first()[0]
93+
).all()[0][0]
9494
== 2023
9595
)
9696
assert (
9797
alembic_connection.execute(
9898
sa.text(f"SELECT promo FROM core_user WHERE user_id = '{user_id_null}'"),
99-
).first()[0]
99+
).all()[0][0]
100100
is None
101101
)

0 commit comments

Comments
 (0)