Coverage for tests / test_missing_type.py: 92%

1from typing import Optional 9 ctx1jklmnopqr

2 

3import pytest 9 ctx1jklmnopqr

4from pydantic import BaseModel 9 ctx1jklmnopqr

5from sqlmodel import Field, SQLModel 9 ctx1jklmnopqr

6 

7 

8def test_missing_sql_type(): 9 ctx1jklmnopqr

9 class CustomType(BaseModel): 9 ctx1abcdefghi

10 @classmethod 9 ctx1abcdefghi

11 def __get_validators__(cls): 9 ctx1abcdefghi

12 yield cls.validate 

13 

14 @classmethod 9 ctx1abcdefghi

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

16 return v 

17 

18 with pytest.raises(ValueError): 9 ctx1abcdefghi

19 

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

21 id: Optional[int] = Field(default=None, primary_key=True) 9 ctx1abcdefghi

22 item: CustomType 9 ctx1abcdefghi