Coverage for tests / test_enums_models.py: 100%

1import enum 24 ctx1abcdefghijklmnopqrstuvwx

2import uuid 24 ctx1abcdefghijklmnopqrstuvwx

3 

4from sqlmodel import Field, SQLModel 24 ctx1abcdefghijklmnopqrstuvwx

5 

6 

7class MyEnum1(str, enum.Enum): 24 ctx1abcdefghijklmnopqrstuvwx

8 A = "A" 24 ctx1abcdefghijklmnopqrstuvwx

9 B = "B" 24 ctx1abcdefghijklmnopqrstuvwx

10 

11 

12class MyEnum2(str, enum.Enum): 24 ctx1abcdefghijklmnopqrstuvwx

13 C = "C" 24 ctx1abcdefghijklmnopqrstuvwx

14 D = "D" 24 ctx1abcdefghijklmnopqrstuvwx

15 

16 

17class BaseModel(SQLModel): 24 ctx1abcdefghijklmnopqrstuvwx

18 id: uuid.UUID = Field(primary_key=True) 24 ctx1abcdefghijklmnopqrstuvwx

19 enum_field: MyEnum2 24 ctx1abcdefghijklmnopqrstuvwx

20 

21 

22class FlatModel(SQLModel, table=True): 24 ctx1abcdefghijklmnopqrstuvwx

23 id: uuid.UUID = Field(primary_key=True) 24 ctx1abcdefghijklmnopqrstuvwx

24 enum_field: MyEnum1 24 ctx1abcdefghijklmnopqrstuvwx

25 

26 

27class InheritModel(BaseModel, table=True): 24 ctx1abcdefghijklmnopqrstuvwx

28 pass 24 ctx1abcdefghijklmnopqrstuvwx