Coverage for tests / test_missing_type.py: 92%

1import pytest 9 ctx1jklmnopqr

2from pydantic import BaseModel 9 ctx1jklmnopqr

3from sqlmodel import Field, SQLModel 9 ctx1jklmnopqr

4 

5 

6def test_missing_sql_type(): 9 ctx1jklmnopqr

7 class CustomType(BaseModel): 9 ctx1abcdefghi

8 @classmethod 9 ctx1abcdefghi

9 def __get_validators__(cls): 9 ctx1abcdefghi

10 yield cls.validate 

11 

12 @classmethod 9 ctx1abcdefghi

13 def validate(cls, v): # pragma: no cover 9 ctx1abcdefghi

14 return v 

15 

16 with pytest.raises(ValueError): 9 ctx1abcdefghi

17 

18 class Item(SQLModel, table=True): 9 ctx1abcdefghi

19 id: int | None = Field(default=None, primary_key=True) 9 ctx1abcdefghi

20 item: CustomType 9 ctx1abcdefghi