From 5f4d502030c26845b7bc7c73c4de4cce329b44e6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebasti=C3=A1n=20Ram=C3=ADrez?= Date: Mon, 4 Dec 2023 15:21:14 +0100 Subject: [PATCH] =?UTF-8?q?=E2=9C=85=20Update=20tests=20to=20check=20for?= =?UTF-8?q?=20deprecation?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_query.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_query.py b/tests/test_query.py index abca97253..88517b92f 100644 --- 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