✅ 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_one/test_tutorial004.py

@@ -1,17 +1,28 @@
-from unittest.mock import patch
+import importlib
+from types import ModuleType
 
 import pytest
 from sqlalchemy.exc import MultipleResultsFound
 from sqlmodel import Session, create_engine, delete
 
-from ...conftest import get_testing_print_function
+from ...conftest import PrintMock, needs_py310
 
 
-def test_tutorial(clear_sqlmodel):
-    from docs_src.tutorial.one 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.one.{request.param}")
     mod.sqlite_url = "sqlite://"
     mod.engine = create_engine(mod.sqlite_url)
+    return mod
+
+
+def test_tutorial(print_mock: PrintMock, mod: ModuleType):
     with pytest.raises(MultipleResultsFound):
         mod.main()
     with Session(mod.engine) as session:
@@ -21,13 +32,8 @@ def test_tutorial(clear_sqlmodel):
         session.add(mod.Hero(name="Test Hero", secret_name="Secret Test Hero", age=24))
         session.commit()
 
-    calls = []
-
-    new_print = get_testing_print_function(calls)
-
-    with patch("builtins.print", new=new_print):
-        mod.select_heroes()
-    assert calls == [
+    mod.select_heroes()
+    assert print_mock.calls == [
         [
             "Hero:",
             {