Coverage for tests/test_missing_type.py: 100%

1from typing import Optional 6 ctx1ghijkl

2 

3import pytest 6 ctx1ghijkl

4from pydantic import BaseModel 6 ctx1ghijkl

5from sqlmodel import Field, SQLModel 6 ctx1ghijkl

6 

7 

8def test_missing_sql_type(): 6 ctx1ghijkl

9 class CustomType(BaseModel): 6 ctx1abcdef

10 @classmethod 6 ctx1abcdef

11 def __get_validators__(cls): 6 ctx1abcdef

12 yield cls.validate 6 ctx1abcdef

13 

14 @classmethod 6 ctx1abcdef

15 def validate(cls, v): # pragma: no cover 6 ctx1abcdef

16 return v 

17 

18 with pytest.raises(ValueError): 6 ctx1abcdef

19 

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

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

22 item: CustomType 6 ctx1abcdef