Use better assertion in test

fastapi · Aug 7, 2024 · dfd9b30 · dfd9b30
1 parent 23aff38
commit dfd9b30
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/test_validation.py b/tests/test_validation.py
@@ -125,10 +125,8 @@ class Hero(SQLModel, table=True):
 
     with Session(engine) as session:
         hero = session.get(Hero, 1)
-        assert not session.dirty
-        assert not session.new
+        assert session._is_clean()
 
         Hero.model_validate(hero)
 
-        assert not session.dirty
-        assert not session.new
+        assert session._is_clean()