Coverage for tests/test_missing_type.py: 100%

1from typing import Optional 6 ctx1feabcd

2 

3import pytest 6 ctx1feabcd

4from pydantic import BaseModel 6 ctx1feabcd

5from sqlmodel import Field, SQLModel 6 ctx1feabcd

6 

7 

8def test_missing_sql_type(): 6 ctx1feabcd

9 class CustomType(BaseModel): 6 ctx1feabcd

10 @classmethod 6 ctx1feabcd

11 def __get_validators__(cls): 5 ctx1eabcd

12 yield cls.validate 4 ctx1abcd

13 

14 @classmethod 6 ctx1feabcd

15 def validate(cls, v): # pragma: no cover 5 ctx1eabcd

16 return v 

17 

18 with pytest.raises(ValueError): 6 ctx1feabcd

19 

20 class Item(SQLModel, table=True): 6 ctx1feabcd

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

22 item: CustomType 6 ctx1feabcd