sqlmodel-python/tests/test_fields_set.py at main · mz0in/sqlmodel-python · GitHub

Repository navigation

21 lines (17 loc) · 654 Bytes

from datetime import datetime, timedelta

from sqlmodel import Field, SQLModel

def test_fields_set():

class User(SQLModel):

username: str

email: str = "test@test.com"

last_updated: datetime = Field(default_factory=datetime.now)

user = User(username="bob")

assert user.__fields_set__ == {"username"}

user = User(username="bob", email="bob@test.com")

assert user.__fields_set__ == {"username", "email"}

user = User(

username="bob",

email="bob@test.com",

last_updated=datetime.now() - timedelta(days=1),

)

assert user.__fields_set__ == {"username", "email", "last_updated"}