Merge pull request #1190 from effigies/mnt/setuptools-distutils-purge

MNT: Finish removing distutils and setuptools dependencies

Author: effigies

.pre-commit-config.yaml

@@ -37,5 +37,6 @@ repos:
           - pydicom
           - numpy
           - pyzstd
+          - importlib_resources
         args: ["nibabel"]
         pass_filenames: false

nibabel/__init__.py

@@ -171,11 +171,16 @@ def bench(label=None, verbose=1, extra_argv=None):
     code : ExitCode
         Returns the result of running the tests as a ``pytest.ExitCode`` enum
     """
-    from pkg_resources import resource_filename
+    try:
+        from importlib.resources import as_file, files
+    except ImportError:
+        from importlib_resources import as_file, files
 
-    config = resource_filename('nibabel', 'benchmarks/pytest.benchmark.ini')
     args = []
     if extra_argv is not None:
         args.extend(extra_argv)
-    args.extend(['-c', config])
-    return test(label, verbose, extra_argv=args)
+
+    config_path = files('nibabel') / 'benchmarks/pytest.benchmark.ini'
+    with as_file(config_path) as config:
+        args.extend(['-c', str(config)])
+        return test(label, verbose, extra_argv=args)

nibabel/cmdline/tests/test_conform.py

@@ -15,15 +15,15 @@
 import nibabel as nib
 from nibabel.cmdline.conform import main
 from nibabel.optpkg import optional_package
-from nibabel.testing import test_data
+from nibabel.testing import get_test_data
 
 _, have_scipy, _ = optional_package('scipy.ndimage')
 needs_scipy = unittest.skipUnless(have_scipy, 'These tests need scipy')
 
 
 @needs_scipy
 def test_default(tmpdir):
-    infile = test_data(fname='anatomical.nii')
+    infile = get_test_data(fname='anatomical.nii')
     outfile = tmpdir / 'output.nii.gz'
     main([str(infile), str(outfile)])
     assert outfile.isfile()
@@ -41,7 +41,7 @@ def test_default(tmpdir):
 
 @needs_scipy
 def test_nondefault(tmpdir):
-    infile = test_data(fname='anatomical.nii')
+    infile = get_test_data(fname='anatomical.nii')
     outfile = tmpdir / 'output.nii.gz'
     out_shape = (100, 100, 150)
     voxel_size = (1, 2, 4)

nibabel/cmdline/tests/test_convert.py

@@ -13,11 +13,11 @@
 
 import nibabel as nib
 from nibabel.cmdline import convert
-from nibabel.testing import test_data
+from nibabel.testing import get_test_data
 
 
 def test_convert_noop(tmp_path):
-    infile = test_data(fname='anatomical.nii')
+    infile = get_test_data(fname='anatomical.nii')
     outfile = tmp_path / 'output.nii.gz'
 
     orig = nib.load(infile)
@@ -31,7 +31,7 @@ def test_convert_noop(tmp_path):
     assert converted.shape == orig.shape
     assert converted.get_data_dtype() == orig.get_data_dtype()
 
-    infile = test_data(fname='resampled_anat_moved.nii')
+    infile = get_test_data(fname='resampled_anat_moved.nii')
 
     with pytest.raises(FileExistsError):
         convert.main([str(infile), str(outfile)])
@@ -50,7 +50,7 @@ def test_convert_noop(tmp_path):
 
 @pytest.mark.parametrize('data_dtype', ('u1', 'i2', 'float32', 'float', 'int64'))
 def test_convert_dtype(tmp_path, data_dtype):
-    infile = test_data(fname='anatomical.nii')
+    infile = get_test_data(fname='anatomical.nii')
     outfile = tmp_path / 'output.nii.gz'
 
     orig = nib.load(infile)
@@ -78,7 +78,7 @@ def test_convert_dtype(tmp_path, data_dtype):
     ],
 )
 def test_convert_by_extension(tmp_path, ext, img_class):
-    infile = test_data(fname='anatomical.nii')
+    infile = get_test_data(fname='anatomical.nii')
     outfile = tmp_path / f'output.{ext}'
 
     orig = nib.load(infile)
@@ -102,7 +102,7 @@ def test_convert_by_extension(tmp_path, ext, img_class):
     ],
 )
 def test_convert_imgtype(tmp_path, ext, img_class):
-    infile = test_data(fname='anatomical.nii')
+    infile = get_test_data(fname='anatomical.nii')
     outfile = tmp_path / f'output.{ext}'
 
     orig = nib.load(infile)
@@ -118,7 +118,7 @@ def test_convert_imgtype(tmp_path, ext, img_class):
 
 
 def test_convert_nifti_int_fail(tmp_path):
-    infile = test_data(fname='anatomical.nii')
+    infile = get_test_data(fname='anatomical.nii')
     outfile = tmp_path / f'output.nii'
 
     orig = nib.load(infile)

nibabel/gifti/gifti.py

@@ -701,8 +701,8 @@ def agg_data(self, intent_code=None):
         Consider a surface GIFTI file:
 
         >>> import nibabel as nib
-        >>> from nibabel.testing import test_data
-        >>> surf_img = nib.load(test_data('gifti', 'ascii.gii'))
+        >>> from nibabel.testing import get_test_data
+        >>> surf_img = nib.load(get_test_data('gifti', 'ascii.gii'))
 
         The coordinate data, which is indicated by the ``NIFTI_INTENT_POINTSET``
         intent code, may be retrieved using any of the following equivalent
@@ -754,7 +754,7 @@ def agg_data(self, intent_code=None):
         The following image is a GIFTI file with ten (10) data arrays of the same
         size, and with intent code 2001 (``NIFTI_INTENT_TIME_SERIES``):
 
-        >>> func_img = nib.load(test_data('gifti', 'task.func.gii'))
+        >>> func_img = nib.load(get_test_data('gifti', 'task.func.gii'))
 
         When aggregating time series data, these arrays are concatenated into
         a single, vertex-by-timestep array:

nibabel/gifti/tests/test_gifti.py

@@ -14,7 +14,7 @@
 from ... import load
 from ...fileholders import FileHolder
 from ...nifti1 import data_type_codes
-from ...testing import test_data
+from ...testing import get_test_data
 from .. import (
     GiftiCoordSystem,
     GiftiDataArray,
@@ -35,9 +35,9 @@
 
 
 def test_agg_data():
-    surf_gii_img = load(test_data('gifti', 'ascii.gii'))
-    func_gii_img = load(test_data('gifti', 'task.func.gii'))
-    shape_gii_img = load(test_data('gifti', 'rh.shape.curv.gii'))
+    surf_gii_img = load(get_test_data('gifti', 'ascii.gii'))
+    func_gii_img = load(get_test_data('gifti', 'task.func.gii'))
+    shape_gii_img = load(get_test_data('gifti', 'rh.shape.curv.gii'))
     # add timeseries data with intent code ``none``
 
     point_data = surf_gii_img.get_arrays_from_intent('pointset')[0].data
@@ -478,7 +478,7 @@ def test_darray_dtype_coercion_failures():
 
 
 def test_gifti_file_close(recwarn):
-    gii = load(test_data('gifti', 'ascii.gii'))
+    gii = load(get_test_data('gifti', 'ascii.gii'))
     with InTemporaryDirectory():
         gii.to_filename('test.gii')
     assert not any(isinstance(r.message, ResourceWarning) for r in recwarn)

nibabel/pkg_info.py

@@ -101,7 +101,7 @@ def pkg_commit_hash(pkg_path: str | None = None) -> tuple[str, str]:
         return 'archive substitution', COMMIT_HASH
     ver = Version(__version__)
     if ver.local is not None and ver.local.startswith('g'):
-        return ver.local[1:8], 'installation'
+        return 'installation', ver.local[1:8]
     # maybe we are in a repository
     proc = run(
         ('git', 'rev-parse', '--short', 'HEAD'),

nibabel/testing/__init__.py

@@ -7,10 +7,12 @@
 #
 ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ### ##
 """Utilities for testing"""
+from __future__ import annotations
 
 import os
 import re
 import sys
+import typing as ty
 import unittest
 import warnings
 from contextlib import nullcontext
@@ -19,28 +21,38 @@
 import numpy as np
 import pytest
 from numpy.testing import assert_array_equal
-from pkg_resources import resource_filename
 
 from .helpers import assert_data_similar, bytesio_filemap, bytesio_round_trip
 from .np_features import memmap_after_ufunc
 
+try:
+    from importlib.abc import Traversable
+    from importlib.resources import as_file, files
+except ImportError:  # PY38
+    from importlib_resources import as_file, files
+    from importlib_resources.abc import Traversable
 
-def test_data(subdir=None, fname=None):
+
+def get_test_data(
+    subdir: ty.Literal['gifti', 'nicom', 'externals'] | None = None,
+    fname: str | None = None,
+) -> Traversable:
+    parts: tuple[str, ...]
     if subdir is None:
-        resource = os.path.join('tests', 'data')
+        parts = ('tests', 'data')
     elif subdir in ('gifti', 'nicom', 'externals'):
-        resource = os.path.join(subdir, 'tests', 'data')
+        parts = (subdir, 'tests', 'data')
     else:
         raise ValueError(f'Unknown test data directory: {subdir}')
 
     if fname is not None:
-        resource = os.path.join(resource, fname)
+        parts += (fname,)
 
-    return resource_filename('nibabel', resource)
+    return files('nibabel').joinpath(*parts)
 
 
 # set path to example data
-data_path = test_data()
+data_path = get_test_data()
 
 
 def assert_dt_equal(a, b):

nibabel/tests/test_casting.py

@@ -233,10 +233,15 @@ def test_best_float():
 
 
 def test_longdouble_precision_improved():
-    # Just check that this can only be True on windows, msvc
-    from numpy.distutils.ccompiler import get_default_compiler
+    # Just check that this can only be True on Windows
 
-    if not (os.name == 'nt' and get_default_compiler() == 'msvc'):
+    # This previously used distutils.ccompiler.get_default_compiler to check for msvc
+    # In https://github.com/python/cpython/blob/3467991/Lib/distutils/ccompiler.py#L919-L956
+    # we see that this was implied by os.name == 'nt', so we can remove this deprecated
+    # call.
+    # However, there may be detectable conditions in Windows where we would expect this
+    # to be False as well.
+    if os.name != 'nt':
         assert not longdouble_precision_improved()
 
 

nibabel/tests/test_init.py

@@ -1,7 +1,12 @@
+import pathlib
 from unittest import mock
 
 import pytest
-from pkg_resources import resource_filename
+
+try:
+    from importlib.resources import as_file, files
+except ImportError:
+    from importlib_resources import as_file, files
 
 import nibabel as nib
 
@@ -38,12 +43,11 @@ def test_nibabel_test_errors():
 
 
 def test_nibabel_bench():
-    expected_args = ['-c', '--pyargs', 'nibabel']
+    config_path = files('nibabel') / 'benchmarks/pytest.benchmark.ini'
+    if not isinstance(config_path, pathlib.Path):
+        raise unittest.SkipTest('Package is not unpacked; could get temp path')
 
-    try:
-        expected_args.insert(1, resource_filename('nibabel', 'benchmarks/pytest.benchmark.ini'))
-    except:
-        raise unittest.SkipTest('Not installed')
+    expected_args = ['-c', str(config_path), '--pyargs', 'nibabel']
 
     with mock.patch('pytest.main') as pytest_main:
         nib.bench(verbose=0)