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

1from decimal import Decimal 10 ctx1uvwxyzABCD

2from typing import Optional, Union 10 ctx1uvwxyzABCD

3 

4import pytest 10 ctx1uvwxyzABCD

5from pydantic import ValidationError 10 ctx1uvwxyzABCD

6from sqlmodel import Field, SQLModel 10 ctx1uvwxyzABCD

7from typing_extensions import Literal 10 ctx1uvwxyzABCD

8 

9 

10def test_decimal(): 10 ctx1uvwxyzABCD

11 class Model(SQLModel): 10 ctx1klmnopqrst

12 dec: Decimal = Field(max_digits=4, decimal_places=2) 10 ctx1klmnopqrst

13 

14 Model(dec=Decimal("3.14")) 10 ctx1klmnopqrst

15 Model(dec=Decimal("69.42")) 10 ctx1klmnopqrst

16 

17 with pytest.raises(ValidationError): 10 ctx1klmnopqrst

18 Model(dec=Decimal("3.142")) 10 ctx1klmnopqrst

19 with pytest.raises(ValidationError): 10 ctx1klmnopqrst

20 Model(dec=Decimal("0.069")) 10 ctx1klmnopqrst

21 with pytest.raises(ValidationError): 10 ctx1klmnopqrst

22 Model(dec=Decimal("420")) 10 ctx1klmnopqrst

23 

24 

25def test_discriminator(): 10 ctx1uvwxyzABCD

26 # Example adapted from 

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

28 

29 class Cat(SQLModel): 10 ctx1abcdefghij

30 pet_type: Literal["cat"] 10 ctx1abcdefghij

31 meows: int 10 ctx1abcdefghij

32 

33 class Dog(SQLModel): 10 ctx1abcdefghij

34 pet_type: Literal["dog"] 10 ctx1abcdefghij

35 barks: float 10 ctx1abcdefghij

36 

37 class Lizard(SQLModel): 10 ctx1abcdefghij

38 pet_type: Literal["reptile", "lizard"] 10 ctx1abcdefghij

39 scales: bool 10 ctx1abcdefghij

40 

41 class Model(SQLModel): 10 ctx1abcdefghij

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

43 n: int 10 ctx1abcdefghij

44 

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

46 

47 with pytest.raises(ValidationError): 10 ctx1abcdefghij

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

49 

50 

51def test_repr(): 10 ctx1uvwxyzABCD

52 class Model(SQLModel): 10 ctx1EFGHIJKLMN

53 id: Optional[int] = Field(primary_key=True) 10 ctx1EFGHIJKLMN

54 foo: str = Field(repr=False) 10 ctx1EFGHIJKLMN

55 

56 instance = Model(id=123, foo="bar") 10 ctx1EFGHIJKLMN

57 assert "foo=" not in repr(instance) 10 ctx1EFGHIJKLMN