sqlmodel/tests/test_missing_type.py at main · EndexAI/sqlmodel · GitHub

Repository navigation

22 lines (16 loc) · 536 Bytes

from typing import Optional

import pytest

from pydantic import BaseModel

from sqlmodel import Field, SQLModel

def test_missing_sql_type():

class CustomType(BaseModel):

@classmethod

def __get_validators__(cls):

yield cls.validate

@classmethod

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

return v

with pytest.raises(ValueError):

class Item(SQLModel, table=True):

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

item: CustomType