Coverage for tests/test_tutorial/test_limit_and_offset/test_tutorial002_py310.py: 100%

1from unittest.mock import patch 6 ctx1efghij

2 

3from sqlmodel import create_engine 6 ctx1efghij

4 

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

6 

7expected_calls = [ 6 ctx1efghij

8 [ 

9 [ 

10 { 

11 "id": 4, 

12 "name": "Tarantula", 

13 "secret_name": "Natalia Roman-on", 

14 "age": 32, 

15 }, 

16 {"id": 5, "name": "Black Lion", "secret_name": "Trevor Challa", "age": 35}, 

17 {"id": 6, "name": "Dr. Weird", "secret_name": "Steve Weird", "age": 36}, 

18 ] 

19 ] 

20] 

21 

22 

23@needs_py310 6 ctx1efghij

24def test_tutorial(clear_sqlmodel): 6 ctx1efghij

25 from docs_src.tutorial.offset_and_limit import tutorial002_py310 as mod 4 ctx1abcd

26 

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

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

29 calls = [] 4 ctx1abcd

30 

31 new_print = get_testing_print_function(calls) 4 ctx1abcd

32 

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

34 mod.main() 4 ctx1abcd

35 assert calls == expected_calls 4 ctx1abcd