Skip to content

Commit a0ec151

Browse files
committed
♻️ Refactor test to not depend on a util for it
1 parent aad9f7c commit a0ec151

1 file changed

Lines changed: 3 additions & 4 deletions

File tree

tests/test_fields_set.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
from datetime import datetime, timedelta
22

33
from sqlmodel import Field, SQLModel
4-
from sqlmodel._compat import get_fields_set
54

65

76
def test_fields_set():
@@ -11,12 +10,12 @@ class User(SQLModel):
1110
last_updated: datetime = Field(default_factory=datetime.now)
1211

1312
user = User(username="bob")
14-
assert get_fields_set(user) == {"username"}
13+
assert user.model_fields_set == {"username"}
1514
user = User(username="bob", email="bob@test.com")
16-
assert get_fields_set(user) == {"username", "email"}
15+
assert user.model_fields_set == {"username", "email"}
1716
user = User(
1817
username="bob",
1918
email="bob@test.com",
2019
last_updated=datetime.now() - timedelta(days=1),
2120
)
22-
assert get_fields_set(user) == {"username", "email", "last_updated"}
21+
assert user.model_fields_set == {"username", "email", "last_updated"}

0 commit comments

Comments
 (0)