Coverage for tests/test_advanced/test_decimal/test_tutorial001_py310.py: 100%

1from decimal import Decimal 6 ctx1efghij

2from unittest.mock import patch 6 ctx1efghij

3 

4from sqlmodel import create_engine 6 ctx1efghij

5 

6from ...conftest import get_testing_print_function, needs_py310 6 ctx1efghij

7 

8expected_calls = [ 6 ctx1efghij

9 [ 

10 "Hero 1:", 

11 { 

12 "name": "Deadpond", 

13 "age": None, 

14 "id": 1, 

15 "secret_name": "Dive Wilson", 

16 "money": Decimal("1.100"), 

17 }, 

18 ], 

19 [ 

20 "Hero 2:", 

21 { 

22 "name": "Rusty-Man", 

23 "age": 48, 

24 "id": 3, 

25 "secret_name": "Tommy Sharp", 

26 "money": Decimal("2.200"), 

27 }, 

28 ], 

29 ["Total money: 3.300"], 

30] 

31 

32 

33@needs_py310 6 ctx1efghij

34def test_tutorial(clear_sqlmodel): 6 ctx1efghij

35 from docs_src.advanced.decimal import tutorial001_py310 as mod 4 ctx1abcd

36 

37 mod.sqlite_url = "sqlite://" 4 ctx1abcd

38 mod.engine = create_engine(mod.sqlite_url) 4 ctx1abcd

39 calls = [] 4 ctx1abcd

40 

41 new_print = get_testing_print_function(calls) 4 ctx1abcd

42 

43 with patch("builtins.print", new=new_print): 4 ctx1abcd

44 mod.main() 4 ctx1abcd

45 assert calls == expected_calls 4 ctx1abcd