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

1from decimal import Decimal 9 ctx1stuvwxyzA

2from typing import Optional, Union 9 ctx1stuvwxyzA

3 

4import pytest 9 ctx1stuvwxyzA

5from pydantic import ValidationError 9 ctx1stuvwxyzA

6from sqlmodel import Field, SQLModel 9 ctx1stuvwxyzA

7from typing_extensions import Literal 9 ctx1stuvwxyzA

8 

9 

10def test_decimal(): 9 ctx1stuvwxyzA

11 class Model(SQLModel): 9 ctx1jklmnopqr

12 dec: Decimal = Field(max_digits=4, decimal_places=2) 9 ctx1jklmnopqr

13 

14 Model(dec=Decimal("3.14")) 9 ctx1jklmnopqr

15 Model(dec=Decimal("69.42")) 9 ctx1jklmnopqr

16 

17 with pytest.raises(ValidationError): 9 ctx1jklmnopqr

18 Model(dec=Decimal("3.142")) 9 ctx1jklmnopqr

19 with pytest.raises(ValidationError): 9 ctx1jklmnopqr

20 Model(dec=Decimal("0.069")) 9 ctx1jklmnopqr

21 with pytest.raises(ValidationError): 9 ctx1jklmnopqr

22 Model(dec=Decimal("420")) 9 ctx1jklmnopqr

23 

24 

25def test_discriminator(): 9 ctx1stuvwxyzA

26 # Example adapted from 

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

28 

29 class Cat(SQLModel): 9 ctx1abcdefghi

30 pet_type: Literal["cat"] 9 ctx1abcdefghi

31 meows: int 9 ctx1abcdefghi

32 

33 class Dog(SQLModel): 9 ctx1abcdefghi

34 pet_type: Literal["dog"] 9 ctx1abcdefghi

35 barks: float 9 ctx1abcdefghi

36 

37 class Lizard(SQLModel): 9 ctx1abcdefghi

38 pet_type: Literal["reptile", "lizard"] 9 ctx1abcdefghi

39 scales: bool 9 ctx1abcdefghi

40 

41 class Model(SQLModel): 9 ctx1abcdefghi

42 pet: Union[Cat, Dog, Lizard] = Field(..., discriminator="pet_type") 9 ctx1abcdefghi

43 n: int 9 ctx1abcdefghi

44 

45 Model(pet={"pet_type": "dog", "barks": 3.14}, n=1) # type: ignore[arg-type] 9 ctx1abcdefghi

46 

47 with pytest.raises(ValidationError): 9 ctx1abcdefghi

48 Model(pet={"pet_type": "dog"}, n=1) # type: ignore[arg-type] 9 ctx1abcdefghi

49 

50 

51def test_repr(): 9 ctx1stuvwxyzA

52 class Model(SQLModel): 9 ctx1BCDEFGHIJ

53 id: Optional[int] = Field(primary_key=True) 9 ctx1BCDEFGHIJ

54 foo: str = Field(repr=False) 9 ctx1BCDEFGHIJ

55 

56 instance = Model(id=123, foo="bar") 9 ctx1BCDEFGHIJ

57 assert "foo=" not in repr(instance) 9 ctx1BCDEFGHIJ