Set globals in pytests

tobiasdiez · Dec 16, 2024 · 678142d · 678142d
1 parent ddba82f
commit 678142d
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/conftest.py b/src/conftest.py
@@ -262,14 +262,17 @@ def doctest_run(
     out: Any = None,
     clear_globs: bool = True,
 ) -> doctest.TestResults:
+    from sage.repl.rich_output import get_display_manager
+    from sage.repl.user_globals import set_globals
+
     traceback.format_exception_only = format_exception_only
 
     # Display warnings in doctests
     warnings.showwarning = showwarning_with_traceback
-
-    from sage.repl.rich_output import get_display_manager
-
     setattr(sys, "__displayhook__", get_display_manager().displayhook)
+
+    # Ensure that injecting globals works as expected in doctests
+    set_globals(test.globs)
     return old_run(self, test, compileflags, out, clear_globs)