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

1from decimal import Decimal 6 ctx1abcdef

2from typing import Optional, Union 6 ctx1abcdef

3 

4import pytest 6 ctx1abcdef

5from pydantic import ValidationError 6 ctx1abcdef

6from sqlmodel import Field, SQLModel 6 ctx1abcdef

7from typing_extensions import Literal 6 ctx1abcdef

8 

9 

10def test_decimal(): 6 ctx1abcdef

11 class Model(SQLModel): 6 ctx1abcdef

12 dec: Decimal = Field(max_digits=4, decimal_places=2) 6 ctx1abcdef

13 

14 Model(dec=Decimal("3.14")) 6 ctx1abcdef

15 Model(dec=Decimal("69.42")) 6 ctx1abcdef

16 

17 with pytest.raises(ValidationError): 6 ctx1abcdef

18 Model(dec=Decimal("3.142")) 6 ctx1abcdef

19 with pytest.raises(ValidationError): 6 ctx1abcdef

20 Model(dec=Decimal("0.069")) 6 ctx1abcdef

21 with pytest.raises(ValidationError): 6 ctx1abcdef

22 Model(dec=Decimal("420")) 6 ctx1abcdef

23 

24 

25def test_discriminator(): 6 ctx1abcdef

26 # Example adapted from 

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

28 

29 class Cat(SQLModel): 6 ctx1abcdef

30 pet_type: Literal["cat"] 6 ctx1abcdef

31 meows: int 6 ctx1abcdef

32 

33 class Dog(SQLModel): 6 ctx1abcdef

34 pet_type: Literal["dog"] 6 ctx1abcdef

35 barks: float 6 ctx1abcdef

36 

37 class Lizard(SQLModel): 6 ctx1abcdef

38 pet_type: Literal["reptile", "lizard"] 6 ctx1abcdef

39 scales: bool 6 ctx1abcdef

40 

41 class Model(SQLModel): 6 ctx1abcdef

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

43 n: int 6 ctx1abcdef

44 

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

46 

47 with pytest.raises(ValidationError): 6 ctx1abcdef

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

49 

50 

51def test_repr(): 6 ctx1abcdef

52 class Model(SQLModel): 6 ctx1abcdef

53 id: Optional[int] = Field(primary_key=True) 6 ctx1abcdef

54 foo: str = Field(repr=False) 6 ctx1abcdef

55 

56 instance = Model(id=123, foo="bar") 6 ctx1abcdef

57 assert "foo=" not in repr(instance) 6 ctx1abcdef