src/sage/repl/display/formatter.py: Replace use of SAGE_EXTCODE in do…

…ctest by importlib.resources
sagemath · Dec 6, 2021 · 0f63cce · 0f63cce
1 parent b2cb173
commit 0f63cce
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/sage/repl/display/formatter.py b/src/sage/repl/display/formatter.py
@@ -140,16 +140,16 @@ def format(self, obj, include=None, exclude=None):
         TESTS::
 
             sage: import os
-            sage: from sage.env import SAGE_EXTCODE
-            sage: example_png = os.path.join(SAGE_EXTCODE, 'doctest', 'rich_output', 'example.png')
+            sage: import importlib.resources
             sage: from sage.repl.rich_output.backend_ipython import BackendIPython
             sage: backend = BackendIPython()
             sage: shell = get_test_shell()
             sage: backend.install(shell=shell)
             sage: shell.run_cell('get_ipython().display_formatter')
             <sage.repl.display.formatter.SageDisplayFormatter object at 0x...>
             sage: shell.run_cell('from IPython.display import Image')
-            sage: shell.run_cell('ipython_image = Image("{0}")'.format(example_png))
+            sage: with importlib.resources.path(sage.repl.rich_output, 'example.png') as example_png:
+            ....:     shell.run_cell('ipython_image = Image("{0}")'.format(example_png))
             sage: shell.run_cell('ipython_image')
             <IPython.core.display.Image object>
             sage: shell.run_cell('get_ipython().display_formatter.format(ipython_image)')