✅ Update tests to use compat get_fields_set

fastapi · Feb 16, 2024 · 15765d4 · 15765d4
1 parent 0c3d2bd
commit 15765d4
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_fields_set.py b/tests/test_fields_set.py
@@ -1,6 +1,7 @@
 from datetime import datetime, timedelta
 
 from sqlmodel import Field, SQLModel
+from sqlmodel._compat import get_fields_set
 
 
 def test_fields_set():
@@ -10,12 +11,12 @@ class User(SQLModel):
         last_updated: datetime = Field(default_factory=datetime.now)
 
     user = User(username="bob")
-    assert user.__fields_set__ == {"username"}
+    assert get_fields_set(user) == {"username"}
     user = User(username="bob", email="[email protected]")
-    assert user.__fields_set__ == {"username", "email"}
+    assert get_fields_set(user) == {"username", "email"}
     user = User(
         username="bob",
         email="[email protected]",
         last_updated=datetime.now() - timedelta(days=1),
     )
-    assert user.__fields_set__ == {"username", "email", "last_updated"}
+    assert get_fields_set(user) == {"username", "email", "last_updated"}