Coverage for tests / test_update.py: 100%

1from sqlmodel import Field, SQLModel 8 ctx1ijklmnop

2 

3 

4def test_sqlmodel_update(): 8 ctx1ijklmnop

5 class Organization(SQLModel, table=True): 8 ctx1abcdefgh

6 id: int = Field(default=None, primary_key=True) 8 ctx1abcdefgh

7 name: str 8 ctx1abcdefgh

8 headquarters: str 8 ctx1abcdefgh

9 

10 class OrganizationUpdate(SQLModel): 8 ctx1abcdefgh

11 name: str 8 ctx1abcdefgh

12 

13 org = Organization(name="Example Org", city="New York", headquarters="NYC HQ") 8 ctx1abcdefgh

14 org_in = OrganizationUpdate(name="Updated org") 8 ctx1abcdefgh

15 org.sqlmodel_update( 8 ctx1abcdefgh

16 org_in, 

17 update={ 

18 "headquarters": "-", # This field is in Organization, but not in OrganizationUpdate 

19 }, 

20 )