✅ Update tests to check for deprecation

tiangolo · Dec 4, 2023 · 5f4d502 · 5f4d502
1 parent 3b2f955
commit 5f4d502
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_query.py b/tests/test_query.py
@@ -1,5 +1,6 @@
 from typing import Optional
 
+import pytest
 from sqlmodel import Field, Session, SQLModel, create_engine
 
 
@@ -21,6 +22,7 @@ class Hero(SQLModel, table=True):
         session.refresh(hero_1)
 
     with Session(engine) as session:
-        query_hero = session.query(Hero).first()
+        with pytest.warns(DeprecationWarning):
+            query_hero = session.query(Hero).first()
         assert query_hero
         assert query_hero.name == hero_1.name