Coverage for tests / test_pydantic / test_field.py: 100%

1from decimal import Decimal 8 ctx1qrstuvwx

2from typing import Literal 8 ctx1qrstuvwx

3 

4import pytest 8 ctx1qrstuvwx

5from pydantic import ValidationError 8 ctx1qrstuvwx

6from sqlmodel import Field, SQLModel 8 ctx1qrstuvwx

7 

8 

9def test_decimal(): 8 ctx1qrstuvwx

10 class Model(SQLModel): 8 ctx1ijklmnop

11 dec: Decimal = Field(max_digits=4, decimal_places=2) 8 ctx1ijklmnop

12 

13 Model(dec=Decimal("3.14")) 8 ctx1ijklmnop

14 Model(dec=Decimal("69.42")) 8 ctx1ijklmnop

15 

16 with pytest.raises(ValidationError): 8 ctx1ijklmnop

17 Model(dec=Decimal("3.142")) 8 ctx1ijklmnop

18 with pytest.raises(ValidationError): 8 ctx1ijklmnop

19 Model(dec=Decimal("0.069")) 8 ctx1ijklmnop

20 with pytest.raises(ValidationError): 8 ctx1ijklmnop

21 Model(dec=Decimal("420")) 8 ctx1ijklmnop

22 

23 

24def test_discriminator(): 8 ctx1qrstuvwx

25 # Example adapted from 

26 # [Pydantic docs](https://pydantic-docs.helpmanual.io/usage/types/#discriminated-unions-aka-tagged-unions): 

27 

28 class Cat(SQLModel): 8 ctx1abcdefgh

29 pet_type: Literal["cat"] 8 ctx1abcdefgh

30 meows: int 8 ctx1abcdefgh

31 

32 class Dog(SQLModel): 8 ctx1abcdefgh

33 pet_type: Literal["dog"] 8 ctx1abcdefgh

34 barks: float 8 ctx1abcdefgh

35 

36 class Lizard(SQLModel): 8 ctx1abcdefgh

37 pet_type: Literal["reptile", "lizard"] 8 ctx1abcdefgh

38 scales: bool 8 ctx1abcdefgh

39 

40 class Model(SQLModel): 8 ctx1abcdefgh

41 pet: Cat | Dog | Lizard = Field(..., discriminator="pet_type") 8 ctx1abcdefgh

42 n: int 8 ctx1abcdefgh

43 

44 Model(pet={"pet_type": "dog", "barks": 3.14}, n=1) # type: ignore[arg-type] 8 ctx1abcdefgh

45 

46 with pytest.raises(ValidationError): 8 ctx1abcdefgh

47 Model(pet={"pet_type": "dog"}, n=1) # type: ignore[arg-type] 8 ctx1abcdefgh

48 

49 

50def test_repr(): 8 ctx1qrstuvwx

51 class Model(SQLModel): 8 ctx1yzABCDEF

52 id: int | None = Field(primary_key=True) 8 ctx1yzABCDEF

53 foo: str = Field(repr=False) 8 ctx1yzABCDEF

54 

55 instance = Model(id=123, foo="bar") 8 ctx1yzABCDEF

56 assert "foo=" not in repr(instance) 8 ctx1yzABCDEF