Coverage for tests/test_missing_type.py: 100%

1from typing import Optional 6 ctx1facdeb

2 

3import pytest 6 ctx1facdeb

4from pydantic import BaseModel 6 ctx1facdeb

5from sqlmodel import Field, SQLModel 6 ctx1facdeb

6 

7 

8def test_missing_sql_type(): 6 ctx1facdeb

9 class CustomType(BaseModel): 6 ctx1facdeb

10 @classmethod 6 ctx1facdeb

11 def __get_validators__(cls): 5 ctx1acdeb

12 yield cls.validate 2 ctx1ab

13 

14 @classmethod 6 ctx1facdeb

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

16 return v 

17 

18 with pytest.raises(ValueError): 6 ctx1facdeb

19 

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

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

22 item: CustomType 6 ctx1facdeb