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 ctx1fabcde

10 @classmethod 6 ctx1fabcde

11 def __get_validators__(cls): 5 ctx1abcde

12 yield cls.validate 6 ctx1fabcde

13 

14 @classmethod 6 ctx1fabcde

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

16 return v 

17 

18 with pytest.raises(ValueError): 6 ctx1fabcde

19 

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

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

22 item: CustomType 6 ctx1fabcde