Coverage for tests / test_missing_type.py: 92%

1import pytest 8 ctx1ijklmnop

2from pydantic import BaseModel 8 ctx1ijklmnop

3from sqlmodel import Field, SQLModel 8 ctx1ijklmnop

4 

5 

6def test_missing_sql_type(): 8 ctx1ijklmnop

7 class CustomType(BaseModel): 8 ctx1abcdefgh

8 @classmethod 8 ctx1abcdefgh

9 def __get_validators__(cls): 8 ctx1abcdefgh

10 yield cls.validate 

11 

12 @classmethod 8 ctx1abcdefgh

13 def validate(cls, v): # pragma: no cover 8 ctx1abcdefgh

14 return v 

15 

16 with pytest.raises(ValueError): 8 ctx1abcdefgh

17 

18 class Item(SQLModel, table=True): 8 ctx1abcdefgh

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

20 item: CustomType 8 ctx1abcdefgh