src/sage_setup/clean.py: Update _find_stale_files doctest to use impo…

…rtlib.metadata.files('sagemath-standard')
sagemath · Oct 3, 2022 · 67ff945 · 67ff945
1 parent ef46de5
commit 67ff945
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/sage_setup/clean.py b/src/sage_setup/clean.py
@@ -86,8 +86,14 @@ def _find_stale_files(site_packages, python_packages, python_modules, ext_module
         sage: from sage_setup.find import find_python_sources, find_extra_files
         sage: python_packages, python_modules, cython_modules = find_python_sources(
         ....:     SAGE_SRC, ['sage', 'sage_setup'])
-        sage: extra_files = list(find_extra_files(SAGE_SRC,
-        ....:     ['sage', 'sage_setup'], cythonized_dir, []).items())
+        sage: extra_files = find_extra_files(SAGE_SRC,
+        ....:     ['sage', 'sage_setup'], cythonized_dir, [])
+        sage: from importlib.metadata import files
+        sage: for f in files('sagemath-standard'):
+        ....:     dir = os.path.dirname(str(f))
+        ....:     extra_files[dir] = extra_files.get(dir, [])
+        ....:     extra_files[dir].append(str(f))
+        sage: extra_files = list(extra_files.items())
         sage: from sage_setup.clean import _find_stale_files
 
     TODO: Also check extension modules::