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)')