Coverage for tests/test_enums_models.py: 100%

1import enum 21 ctx1abcdefghijklmnopqrstu

2import uuid 21 ctx1abcdefghijklmnopqrstu

3 

4from sqlmodel import Field, SQLModel 21 ctx1abcdefghijklmnopqrstu

5 

6 

7class MyEnum1(str, enum.Enum): 21 ctx1abcdefghijklmnopqrstu

8 A = "A" 21 ctx1abcdefghijklmnopqrstu

9 B = "B" 21 ctx1abcdefghijklmnopqrstu

10 

11 

12class MyEnum2(str, enum.Enum): 21 ctx1abcdefghijklmnopqrstu

13 C = "C" 21 ctx1abcdefghijklmnopqrstu

14 D = "D" 21 ctx1abcdefghijklmnopqrstu

15 

16 

17class BaseModel(SQLModel): 21 ctx1abcdefghijklmnopqrstu

18 id: uuid.UUID = Field(primary_key=True) 21 ctx1abcdefghijklmnopqrstu

19 enum_field: MyEnum2 21 ctx1abcdefghijklmnopqrstu

20 

21 

22class FlatModel(SQLModel, table=True): 21 ctx1abcdefghijklmnopqrstu

23 id: uuid.UUID = Field(primary_key=True) 21 ctx1abcdefghijklmnopqrstu

24 enum_field: MyEnum1 21 ctx1abcdefghijklmnopqrstu

25 

26 

27class InheritModel(BaseModel, table=True): 21 ctx1abcdefghijklmnopqrstu

28 pass 21 ctx1abcdefghijklmnopqrstu