✅ Simplify tests for code examples, one test file for multiple variants (#1664)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

tests/test_tutorial/test_where/test_tutorial001.py

@@ -1,22 +1,29 @@
-from unittest.mock import patch
+import importlib
+from types import ModuleType
 
+import pytest
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import PrintMock, needs_py310
 
 
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial001 as mod
-
+@pytest.fixture(
+    name="mod",
+    params=[
+        "tutorial001",
+        pytest.param("tutorial001_py310", marks=needs_py310),
+    ],
+)
+def get_module(request: pytest.FixtureRequest) -> ModuleType:
+    mod = importlib.import_module(f"docs_src.tutorial.where.{request.param}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
-    calls = []
+    return mod
 
-    new_print = get_testing_print_function(calls)
 
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
+def test_tutorial(print_mock: PrintMock, mod: ModuleType):
+    mod.main()
+    assert print_mock.calls == [
         [
             {
                 "name": "Deadpond",

tests/test_tutorial/test_where/test_tutorial001_py310.py

@@ -1,29 +0,0 @@
-from unittest.mock import patch
-
-from sqlmodel import create_engine
-
-from ...conftest import get_testing_print_function, needs_py310
-
-
-@needs_py310
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial001_py310 as mod
-
-    mod.sqlite_url = "sqlite://"
-    mod.engine = create_engine(mod.sqlite_url)
-    calls = []
-
-    new_print = get_testing_print_function(calls)
-
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
-        [
-            {
-                "name": "Deadpond",
-                "secret_name": "Dive Wilson",
-                "age": None,
-                "id": 1,
-            }
-        ]
-    ]

tests/test_tutorial/test_where/test_tutorial002.py

@@ -1,22 +1,29 @@
-from unittest.mock import patch
+import importlib
+from types import ModuleType
 
+import pytest
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import PrintMock, needs_py310
 
 
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial002 as mod
-
+@pytest.fixture(
+    name="mod",
+    params=[
+        "tutorial002",
+        pytest.param("tutorial002_py310", marks=needs_py310),
+    ],
+)
+def get_module(request: pytest.FixtureRequest) -> ModuleType:
+    mod = importlib.import_module(f"docs_src.tutorial.where.{request.param}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
-    calls = []
+    return mod
 
-    new_print = get_testing_print_function(calls)
 
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
+def test_tutorial(print_mock: PrintMock, mod: ModuleType):
+    mod.main()
+    assert print_mock.calls == [
         [
             {
                 "name": "Spider-Boy",

tests/test_tutorial/test_where/test_tutorial002_py310.py

@@ -1,30 +0,0 @@
-from unittest.mock import patch
-
-from sqlmodel import create_engine
-
-from ...conftest import get_testing_print_function, needs_py310
-
-
-@needs_py310
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial002_py310 as mod
-
-    mod.sqlite_url = "sqlite://"
-    mod.engine = create_engine(mod.sqlite_url)
-    calls = []
-
-    new_print = get_testing_print_function(calls)
-
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
-        [
-            {
-                "name": "Spider-Boy",
-                "secret_name": "Pedro Parqueador",
-                "age": None,
-                "id": 2,
-            }
-        ],
-        [{"name": "Rusty-Man", "secret_name": "Tommy Sharp", "age": 48, "id": 3}],
-    ]

tests/test_tutorial/test_where/test_tutorial003.py

@@ -1,21 +1,28 @@
-from unittest.mock import patch
+import importlib
+from types import ModuleType
 
+import pytest
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import PrintMock, needs_py310
 
 
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial003 as mod
-
+@pytest.fixture(
+    name="mod",
+    params=[
+        "tutorial003",
+        pytest.param("tutorial003_py310", marks=needs_py310),
+    ],
+)
+def get_module(request: pytest.FixtureRequest) -> ModuleType:
+    mod = importlib.import_module(f"docs_src.tutorial.where.{request.param}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
-    calls = []
+    return mod
 
-    new_print = get_testing_print_function(calls)
 
-    with patch("builtins.print", new=new_print):
-        mod.main()
+def test_tutorial(print_mock: PrintMock, mod: ModuleType):
+    mod.main()
 
     expected_calls = [
         [{"id": 6, "name": "Dr. Weird", "secret_name": "Steve Weird", "age": 36}],
@@ -29,6 +36,7 @@ def test_tutorial(clear_sqlmodel):
             }
         ],
     ]
+    calls = print_mock.calls
     for call in expected_calls:
         assert call in calls, "This expected item should be in the list"
         # Now that this item was checked, remove it from the list

tests/test_tutorial/test_where/test_tutorial003_py310.py

@@ -1,37 +0,0 @@
-from unittest.mock import patch
-
-from sqlmodel import create_engine
-
-from ...conftest import get_testing_print_function, needs_py310
-
-
-@needs_py310
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial003_py310 as mod
-
-    mod.sqlite_url = "sqlite://"
-    mod.engine = create_engine(mod.sqlite_url)
-    calls = []
-
-    new_print = get_testing_print_function(calls)
-
-    with patch("builtins.print", new=new_print):
-        mod.main()
-
-    expected_calls = [
-        [{"id": 6, "name": "Dr. Weird", "secret_name": "Steve Weird", "age": 36}],
-        [{"id": 3, "name": "Rusty-Man", "secret_name": "Tommy Sharp", "age": 48}],
-        [
-            {
-                "id": 7,
-                "name": "Captain North America",
-                "secret_name": "Esteban Rogelios",
-                "age": 93,
-            }
-        ],
-    ]
-    for call in expected_calls:
-        assert call in calls, "This expected item should be in the list"
-        # Now that this item was checked, remove it from the list
-        calls.pop(calls.index(call))
-    assert len(calls) == 0, "The list should only have the expected items"

tests/test_tutorial/test_where/test_tutorial004.py

@@ -1,21 +1,28 @@
-from unittest.mock import patch
+import importlib
+from types import ModuleType
 
+import pytest
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import PrintMock, needs_py310
 
 
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial004 as mod
-
+@pytest.fixture(
+    name="mod",
+    params=[
+        "tutorial004",
+        pytest.param("tutorial004_py310", marks=needs_py310),
+    ],
+)
+def get_module(request: pytest.FixtureRequest) -> ModuleType:
+    mod = importlib.import_module(f"docs_src.tutorial.where.{request.param}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
-    calls = []
+    return mod
 
-    new_print = get_testing_print_function(calls)
 
-    with patch("builtins.print", new=new_print):
-        mod.main()
+def test_tutorial(print_mock: PrintMock, mod: ModuleType):
+    mod.main()
     expected_calls = [
         [{"id": 5, "name": "Black Lion", "secret_name": "Trevor Challa", "age": 35}],
         [{"id": 6, "name": "Dr. Weird", "secret_name": "Steve Weird", "age": 36}],
@@ -29,6 +36,7 @@ def test_tutorial(clear_sqlmodel):
             }
         ],
     ]
+    calls = print_mock.calls
     for call in expected_calls:
         assert call in calls, "This expected item should be in the list"
         # Now that this item was checked, remove it from the list

tests/test_tutorial/test_where/test_tutorial004_py310.py

@@ -1,37 +0,0 @@
-from unittest.mock import patch
-
-from sqlmodel import create_engine
-
-from ...conftest import get_testing_print_function, needs_py310
-
-
-@needs_py310
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial004_py310 as mod
-
-    mod.sqlite_url = "sqlite://"
-    mod.engine = create_engine(mod.sqlite_url)
-    calls = []
-
-    new_print = get_testing_print_function(calls)
-
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    expected_calls = [
-        [{"id": 5, "name": "Black Lion", "secret_name": "Trevor Challa", "age": 35}],
-        [{"id": 6, "name": "Dr. Weird", "secret_name": "Steve Weird", "age": 36}],
-        [{"id": 3, "name": "Rusty-Man", "secret_name": "Tommy Sharp", "age": 48}],
-        [
-            {
-                "id": 7,
-                "name": "Captain North America",
-                "secret_name": "Esteban Rogelios",
-                "age": 93,
-            }
-        ],
-    ]
-    for call in expected_calls:
-        assert call in calls, "This expected item should be in the list"
-        # Now that this item was checked, remove it from the list
-        calls.pop(calls.index(call))
-    assert len(calls) == 0, "The list should only have the expected items"

tests/test_tutorial/test_where/test_tutorial005.py

@@ -1,21 +1,28 @@
-from unittest.mock import patch
+import importlib
+from types import ModuleType
 
+import pytest
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import PrintMock, needs_py310
 
 
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial005 as mod
-
+@pytest.fixture(
+    name="mod",
+    params=[
+        "tutorial005",
+        pytest.param("tutorial005_py310", marks=needs_py310),
+    ],
+)
+def get_module(request: pytest.FixtureRequest) -> ModuleType:
+    mod = importlib.import_module(f"docs_src.tutorial.where.{request.param}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
-    calls = []
+    return mod
 
-    new_print = get_testing_print_function(calls)
 
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
+def test_tutorial(print_mock: PrintMock, mod: ModuleType):
+    mod.main()
+    assert print_mock.calls == [
         [{"name": "Tarantula", "secret_name": "Natalia Roman-on", "age": 32, "id": 4}]
     ]

tests/test_tutorial/test_where/test_tutorial005_py310.py

@@ -1,22 +0,0 @@
-from unittest.mock import patch
-
-from sqlmodel import create_engine
-
-from ...conftest import get_testing_print_function, needs_py310
-
-
-@needs_py310
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial005_py310 as mod
-
-    mod.sqlite_url = "sqlite://"
-    mod.engine = create_engine(mod.sqlite_url)
-    calls = []
-
-    new_print = get_testing_print_function(calls)
-
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
-        [{"name": "Tarantula", "secret_name": "Natalia Roman-on", "age": 32, "id": 4}]
-    ]

tests/test_tutorial/test_where/test_tutorial006.py

@@ -1,22 +1,29 @@
-from unittest.mock import patch
+import importlib
+from types import ModuleType
 
+import pytest
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import PrintMock, needs_py310
 
 
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial006 as mod
-
+@pytest.fixture(
+    name="mod",
+    params=[
+        "tutorial006",
+        pytest.param("tutorial006_py310", marks=needs_py310),
+    ],
+)
+def get_module(request: pytest.FixtureRequest) -> ModuleType:
+    mod = importlib.import_module(f"docs_src.tutorial.where.{request.param}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
-    calls = []
+    return mod
 
-    new_print = get_testing_print_function(calls)
 
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
+def test_tutorial(print_mock: PrintMock, mod: ModuleType):
+    mod.main()
+    assert print_mock.calls == [
         [{"name": "Tarantula", "secret_name": "Natalia Roman-on", "age": 32, "id": 4}],
         [{"name": "Black Lion", "secret_name": "Trevor Challa", "age": 35, "id": 5}],
     ]

tests/test_tutorial/test_where/test_tutorial006_py310.py

@@ -1,23 +0,0 @@
-from unittest.mock import patch
-
-from sqlmodel import create_engine
-
-from ...conftest import get_testing_print_function, needs_py310
-
-
-@needs_py310
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial006_py310 as mod
-
-    mod.sqlite_url = "sqlite://"
-    mod.engine = create_engine(mod.sqlite_url)
-    calls = []
-
-    new_print = get_testing_print_function(calls)
-
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
-        [{"name": "Tarantula", "secret_name": "Natalia Roman-on", "age": 32, "id": 4}],
-        [{"name": "Black Lion", "secret_name": "Trevor Challa", "age": 35, "id": 5}],
-    ]

tests/test_tutorial/test_where/test_tutorial007.py

@@ -1,22 +1,29 @@
-from unittest.mock import patch
+import importlib
+from types import ModuleType
 
+import pytest
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import PrintMock, needs_py310
 
 
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial007 as mod
-
+@pytest.fixture(
+    name="mod",
+    params=[
+        "tutorial007",
+        pytest.param("tutorial007_py310", marks=needs_py310),
+    ],
+)
+def get_module(request: pytest.FixtureRequest) -> ModuleType:
+    mod = importlib.import_module(f"docs_src.tutorial.where.{request.param}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
-    calls = []
+    return mod
 
-    new_print = get_testing_print_function(calls)
 
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
+def test_tutorial(print_mock: PrintMock, mod: ModuleType):
+    mod.main()
+    assert print_mock.calls == [
         [{"id": 5, "name": "Black Lion", "secret_name": "Trevor Challa", "age": 35}],
         [{"id": 6, "name": "Dr. Weird", "secret_name": "Steve Weird", "age": 36}],
     ]

tests/test_tutorial/test_where/test_tutorial007_py310.py

@@ -1,23 +0,0 @@
-from unittest.mock import patch
-
-from sqlmodel import create_engine
-
-from ...conftest import get_testing_print_function, needs_py310
-
-
-@needs_py310
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial007_py310 as mod
-
-    mod.sqlite_url = "sqlite://"
-    mod.engine = create_engine(mod.sqlite_url)
-    calls = []
-
-    new_print = get_testing_print_function(calls)
-
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
-        [{"id": 5, "name": "Black Lion", "secret_name": "Trevor Challa", "age": 35}],
-        [{"id": 6, "name": "Dr. Weird", "secret_name": "Steve Weird", "age": 36}],
-    ]

tests/test_tutorial/test_where/test_tutorial008.py

@@ -1,22 +1,29 @@
-from unittest.mock import patch
+import importlib
+from types import ModuleType
 
+import pytest
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import PrintMock, needs_py310
 
 
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial008 as mod
-
+@pytest.fixture(
+    name="mod",
+    params=[
+        "tutorial008",
+        pytest.param("tutorial008_py310", marks=needs_py310),
+    ],
+)
+def get_module(request: pytest.FixtureRequest) -> ModuleType:
+    mod = importlib.import_module(f"docs_src.tutorial.where.{request.param}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
-    calls = []
+    return mod
 
-    new_print = get_testing_print_function(calls)
 
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
+def test_tutorial(print_mock: PrintMock, mod: ModuleType):
+    mod.main()
+    assert print_mock.calls == [
         [{"id": 5, "name": "Black Lion", "secret_name": "Trevor Challa", "age": 35}],
         [{"id": 6, "name": "Dr. Weird", "secret_name": "Steve Weird", "age": 36}],
     ]

tests/test_tutorial/test_where/test_tutorial008_py310.py

@@ -1,23 +0,0 @@
-from unittest.mock import patch
-
-from sqlmodel import create_engine
-
-from ...conftest import get_testing_print_function, needs_py310
-
-
-@needs_py310
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial008_py310 as mod
-
-    mod.sqlite_url = "sqlite://"
-    mod.engine = create_engine(mod.sqlite_url)
-    calls = []
-
-    new_print = get_testing_print_function(calls)
-
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
-        [{"id": 5, "name": "Black Lion", "secret_name": "Trevor Challa", "age": 35}],
-        [{"id": 6, "name": "Dr. Weird", "secret_name": "Steve Weird", "age": 36}],
-    ]

tests/test_tutorial/test_where/test_tutorial009.py

@@ -1,22 +1,29 @@
-from unittest.mock import patch
+import importlib
+from types import ModuleType
 
+import pytest
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import PrintMock, needs_py310
 
 
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial009 as mod
-
+@pytest.fixture(
+    name="mod",
+    params=[
+        "tutorial009",
+        pytest.param("tutorial009_py310", marks=needs_py310),
+    ],
+)
+def get_module(request: pytest.FixtureRequest) -> ModuleType:
+    mod = importlib.import_module(f"docs_src.tutorial.where.{request.param}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
-    calls = []
+    return mod
 
-    new_print = get_testing_print_function(calls)
 
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
+def test_tutorial(print_mock: PrintMock, mod: ModuleType):
+    mod.main()
+    assert print_mock.calls == [
         [{"name": "Tarantula", "secret_name": "Natalia Roman-on", "age": 32, "id": 4}],
         [{"name": "Black Lion", "secret_name": "Trevor Challa", "age": 35, "id": 5}],
         [

tests/test_tutorial/test_where/test_tutorial009_py310.py

@@ -1,31 +0,0 @@
-from unittest.mock import patch
-
-from sqlmodel import create_engine
-
-from ...conftest import get_testing_print_function, needs_py310
-
-
-@needs_py310
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial009_py310 as mod
-
-    mod.sqlite_url = "sqlite://"
-    mod.engine = create_engine(mod.sqlite_url)
-    calls = []
-
-    new_print = get_testing_print_function(calls)
-
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
-        [{"name": "Tarantula", "secret_name": "Natalia Roman-on", "age": 32, "id": 4}],
-        [{"name": "Black Lion", "secret_name": "Trevor Challa", "age": 35, "id": 5}],
-        [
-            {
-                "name": "Captain North America",
-                "secret_name": "Esteban Rogelios",
-                "age": 93,
-                "id": 7,
-            }
-        ],
-    ]

tests/test_tutorial/test_where/test_tutorial010.py

@@ -1,22 +1,29 @@
-from unittest.mock import patch
+import importlib
+from types import ModuleType
 
+import pytest
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import PrintMock, needs_py310
 
 
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial010 as mod
-
+@pytest.fixture(
+    name="mod",
+    params=[
+        "tutorial010",
+        pytest.param("tutorial010_py310", marks=needs_py310),
+    ],
+)
+def get_module(request: pytest.FixtureRequest) -> ModuleType:
+    mod = importlib.import_module(f"docs_src.tutorial.where.{request.param}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
-    calls = []
+    return mod
 
-    new_print = get_testing_print_function(calls)
 
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
+def test_tutorial(print_mock: PrintMock, mod: ModuleType):
+    mod.main()
+    assert print_mock.calls == [
         [{"name": "Tarantula", "secret_name": "Natalia Roman-on", "age": 32, "id": 4}],
         [{"name": "Black Lion", "secret_name": "Trevor Challa", "age": 35, "id": 5}],
         [

tests/test_tutorial/test_where/test_tutorial010_py310.py

@@ -1,31 +0,0 @@
-from unittest.mock import patch
-
-from sqlmodel import create_engine
-
-from ...conftest import get_testing_print_function, needs_py310
-
-
-@needs_py310
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial010_py310 as mod
-
-    mod.sqlite_url = "sqlite://"
-    mod.engine = create_engine(mod.sqlite_url)
-    calls = []
-
-    new_print = get_testing_print_function(calls)
-
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    assert calls == [
-        [{"name": "Tarantula", "secret_name": "Natalia Roman-on", "age": 32, "id": 4}],
-        [{"name": "Black Lion", "secret_name": "Trevor Challa", "age": 35, "id": 5}],
-        [
-            {
-                "name": "Captain North America",
-                "secret_name": "Esteban Rogelios",
-                "age": 93,
-                "id": 7,
-            }
-        ],
-    ]

tests/test_tutorial/test_where/test_tutorial011.py

@@ -1,21 +1,28 @@
-from unittest.mock import patch
+import importlib
+from types import ModuleType
 
+import pytest
 from sqlmodel import create_engine
 
-from ...conftest import get_testing_print_function
+from ...conftest import PrintMock, needs_py310
 
 
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial011 as mod
-
+@pytest.fixture(
+    name="mod",
+    params=[
+        "tutorial011",
+        pytest.param("tutorial011_py310", marks=needs_py310),
+    ],
+)
+def get_module(request: pytest.FixtureRequest) -> ModuleType:
+    mod = importlib.import_module(f"docs_src.tutorial.where.{request.param}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
-    calls = []
+    return mod
 
-    new_print = get_testing_print_function(calls)
 
-    with patch("builtins.print", new=new_print):
-        mod.main()
+def test_tutorial(print_mock: PrintMock, mod: ModuleType):
+    mod.main()
     expected_calls = [
         [{"id": 5, "name": "Black Lion", "secret_name": "Trevor Challa", "age": 35}],
         [{"id": 6, "name": "Dr. Weird", "secret_name": "Steve Weird", "age": 36}],
@@ -29,6 +36,7 @@ def test_tutorial(clear_sqlmodel):
             }
         ],
     ]
+    calls = print_mock.calls
     for call in expected_calls:
         assert call in calls, "This expected item should be in the list"
         # Now that this item was checked, remove it from the list

tests/test_tutorial/test_where/test_tutorial011_py310.py

@@ -1,37 +0,0 @@
-from unittest.mock import patch
-
-from sqlmodel import create_engine
-
-from ...conftest import get_testing_print_function, needs_py310
-
-
-@needs_py310
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.where import tutorial011_py310 as mod
-
-    mod.sqlite_url = "sqlite://"
-    mod.engine = create_engine(mod.sqlite_url)
-    calls = []
-
-    new_print = get_testing_print_function(calls)
-
-    with patch("builtins.print", new=new_print):
-        mod.main()
-    expected_calls = [
-        [{"id": 5, "name": "Black Lion", "secret_name": "Trevor Challa", "age": 35}],
-        [{"id": 6, "name": "Dr. Weird", "secret_name": "Steve Weird", "age": 36}],
-        [{"id": 3, "name": "Rusty-Man", "secret_name": "Tommy Sharp", "age": 48}],
-        [
-            {
-                "id": 7,
-                "name": "Captain North America",
-                "secret_name": "Esteban Rogelios",
-                "age": 93,
-            }
-        ],
-    ]
-    for call in expected_calls:
-        assert call in calls, "This expected item should be in the list"
-        # Now that this item was checked, remove it from the list
-        calls.pop(calls.index(call))
-    assert len(calls) == 0, "The list should only have the expected items"