fixes migration

Author: Marc-Andrieu

migrations/versions/40_promo_above_2000.py

@@ -4,7 +4,8 @@
 """
 
 import uuid
-from typing import TYPE_CHECKING, Sequence, Union
+from collections.abc import Sequence
+from typing import TYPE_CHECKING
 
 from app.core.schools.schools_type import SchoolType
 
@@ -14,8 +15,6 @@
 import sqlalchemy as sa
 from alembic import op
 
-from app.types.sqlalchemy import TZDateTime
-
 # revision identifiers, used by Alembic.
 revision: str = "91fadc90f892"
 down_revision: str | None = "2880c583d7f1"
@@ -84,19 +83,19 @@ def test_upgrade(
 ) -> None:
     assert (
         alembic_connection.execute(
-            sa.text(f"SELECT promo FROM core_user WHERE user_id = {user_id_23}"),
-        ).first()[9]
+            sa.text(f"SELECT promo FROM core_user WHERE user_id = '{user_id_23}'"),
+        ).first()[0]
         == 2023
     )
     assert (
         alembic_connection.execute(
-            sa.text(f"SELECT promo FROM core_user WHERE user_id = {user_id_2023}"),
-        ).first()[9]
+            sa.text(f"SELECT promo FROM core_user WHERE user_id = '{user_id_2023}'"),
+        ).first()[0]
         == 2023
     )
     assert (
         alembic_connection.execute(
-            sa.text(f"SELECT promo FROM core_user WHERE user_id = {user_id_null}"),
-        ).first()[9]
+            sa.text(f"SELECT promo FROM core_user WHERE user_id = '{user_id_null}'"),
+        ).first()[0]
         is None
     )