Merge branch 'master' into PGH

Author: effigies

nibabel/batteryrunners.py

@@ -252,7 +252,7 @@ def __str__(self):
     def message(self):
         """formatted message string, including fix message if present"""
         if self.fix_msg:
-            return '; '.join((self.problem_msg, self.fix_msg))
+            return f'{self.problem_msg}; {self.fix_msg}'
         return self.problem_msg
 
     def log_raise(self, logger, error_level=40):

nibabel/cmdline/tests/test_convert.py

@@ -71,7 +71,7 @@ def test_convert_dtype(tmp_path, data_dtype):
 
 
 @pytest.mark.parametrize(
-    'ext,img_class',
+    ('ext', 'img_class'),
     [
         ('mgh', nib.MGHImage),
         ('img', nib.Nifti1Pair),
@@ -94,7 +94,7 @@ def test_convert_by_extension(tmp_path, ext, img_class):
 
 
 @pytest.mark.parametrize(
-    'ext,img_class',
+    ('ext', 'img_class'),
     [
         ('mgh', nib.MGHImage),
         ('img', nib.Nifti1Pair),
@@ -141,7 +141,7 @@ def test_convert_nifti_int_fail(tmp_path):
 
 
 @pytest.mark.parametrize(
-    'orig_dtype,alias,expected_dtype',
+    ('orig_dtype', 'alias', 'expected_dtype'),
     [
         ('int64', 'mask', 'uint8'),
         ('int64', 'compat', 'int32'),

nibabel/cmdline/tests/test_roi.py

@@ -119,7 +119,7 @@ def test_nib_roi(tmp_path, inplace):
 
 
 @pytest.mark.parametrize(
-    'args, errmsg',
+    ('args', 'errmsg'),
     (
         (('-i', '1:1'), 'Cannot take zero-length slice'),
         (('-j', '1::2'), 'Downsampling is not supported'),
@@ -138,12 +138,8 @@ def test_nib_roi_bad_slices(capsys, args, errmsg):
 def test_entrypoint(capsys):
     # Check that we handle missing args as expected
     with mock.patch('sys.argv', ['nib-roi', '--help']):
-        try:
+        with pytest.raises(SystemExit):
             main()
-        except SystemExit:
-            pass
-        else:
-            assert False, 'argparse exits on --help. If changing to another parser, update test.'
     captured = capsys.readouterr()
     assert captured.out.startswith('usage: nib-roi')
 

nibabel/data.py

@@ -87,8 +87,7 @@ def list_files(self, relative=True):
         for base, dirs, files in os.walk(self.base_path):
             if relative:
                 base = base[len(self.base_path) + 1 :]
-            for filename in files:
-                out_list.append(pjoin(base, filename))
+            out_list.extend(pjoin(base, filename) for filename in files)
         return out_list
 
 

nibabel/gifti/gifti.py

@@ -374,7 +374,7 @@ def _to_xml_element(self):
     def print_summary(self):
         print('Dataspace: ', xform_codes.niistring[self.dataspace])
         print('XFormSpace: ', xform_codes.niistring[self.xformspace])
-        print('Affine Transformation Matrix: \n', self.xform)
+        print('Affine Transformation Matrix:\n', self.xform)
 
 
 def _data_tag_element(dataarray, encoding, dtype, ordering):

nibabel/gifti/tests/test_gifti.py

@@ -423,13 +423,14 @@ def test_gifti_coord(capsys):
     gcs.xform = None
     gcs.print_summary()
     captured = capsys.readouterr()
-    assert captured.out == '\n'.join(
-        [
-            'Dataspace:  NIFTI_XFORM_UNKNOWN',
-            'XFormSpace:  NIFTI_XFORM_UNKNOWN',
-            'Affine Transformation Matrix: ',
-            ' None\n',
-        ]
+    assert (
+        captured.out
+        == """\
+Dataspace:  NIFTI_XFORM_UNKNOWN
+XFormSpace:  NIFTI_XFORM_UNKNOWN
+Affine Transformation Matrix:
+ None
+"""
     )
     gcs.to_xml()
 

nibabel/nicom/tests/test_dicomwrappers.py

@@ -429,10 +429,9 @@ def fake_shape_dependents(
 
     class PrintBase:
         def __repr__(self):
-            attr_strs = []
-            for attr in dir(self):
-                if attr[0].isupper():
-                    attr_strs.append(f'{attr}={getattr(self, attr)}')
+            attr_strs = [
+                f'{attr}={getattr(self, attr)}' for attr in dir(self) if attr[0].isupper()
+            ]
             return f"{self.__class__.__name__}({', '.join(attr_strs)})"
 
     class DimIdxSeqElem(pydicom.Dataset):

nibabel/nifti1.py

@@ -325,15 +325,15 @@ class NiftiExtension(ty.Generic[T]):
     """
 
     code: int
-    encoding: ty.Optional[str] = None
+    encoding: str | None = None
     _content: bytes
-    _object: ty.Optional[T] = None
+    _object: T | None = None
 
     def __init__(
         self,
-        code: ty.Union[int, str],
+        code: int | str,
         content: bytes = b'',
-        object: ty.Optional[T] = None,
+        object: T | None = None,
     ) -> None:
         """
         Parameters
@@ -565,9 +565,9 @@ class Nifti1DicomExtension(Nifti1Extension[DicomDataset]):
 
     def __init__(
         self,
-        code: ty.Union[int, str],
-        content: ty.Union[bytes, DicomDataset, None] = None,
-        parent_hdr: ty.Optional[Nifti1Header] = None,
+        code: int | str,
+        content: bytes | DicomDataset | None = None,
+        parent_hdr: Nifti1Header | None = None,
     ) -> None:
         """
         Parameters

nibabel/streamlines/tests/test_streamlines.py

@@ -191,13 +191,13 @@ def test_save_tractogram_file(self):
         trk_file = trk.TrkFile(tractogram)
 
         # No need for keyword arguments.
-        with self.assertRaises(ValueError):
+        with pytest.raises(ValueError):
             nib.streamlines.save(trk_file, 'dummy.trk', header={})
 
         # Wrong extension.
         with pytest.warns(ExtensionWarning, match='extension'):
             trk_file = trk.TrkFile(tractogram)
-            with self.assertRaises(ValueError):
+            with pytest.raises(ValueError):
                 nib.streamlines.save(trk_file, 'dummy.tck', header={})
 
         with InTemporaryDirectory():
@@ -207,7 +207,7 @@ def test_save_tractogram_file(self):
 
     def test_save_empty_file(self):
         tractogram = Tractogram(affine_to_rasmm=np.eye(4))
-        for ext, cls in FORMATS.items():
+        for ext in FORMATS:
             with InTemporaryDirectory():
                 filename = 'streamlines' + ext
                 nib.streamlines.save(tractogram, filename)
@@ -216,7 +216,7 @@ def test_save_empty_file(self):
 
     def test_save_simple_file(self):
         tractogram = Tractogram(DATA['streamlines'], affine_to_rasmm=np.eye(4))
-        for ext, cls in FORMATS.items():
+        for ext in FORMATS:
             with InTemporaryDirectory():
                 filename = 'streamlines' + ext
                 nib.streamlines.save(tractogram, filename)
@@ -262,7 +262,7 @@ def test_save_complex_file(self):
     def test_save_sliced_tractogram(self):
         tractogram = Tractogram(DATA['streamlines'], affine_to_rasmm=np.eye(4))
         original_tractogram = tractogram.copy()
-        for ext, cls in FORMATS.items():
+        for ext in FORMATS:
             with InTemporaryDirectory():
                 filename = 'streamlines' + ext
                 nib.streamlines.save(tractogram[::2], filename)
@@ -272,18 +272,18 @@ def test_save_sliced_tractogram(self):
                 assert_tractogram_equal(tractogram, original_tractogram)
 
     def test_load_unknown_format(self):
-        with self.assertRaises(ValueError):
+        with pytest.raises(ValueError):
             nib.streamlines.load('')
 
     def test_save_unknown_format(self):
-        with self.assertRaises(ValueError):
+        with pytest.raises(ValueError):
             nib.streamlines.save(Tractogram(), '')
 
     def test_save_from_generator(self):
         tractogram = Tractogram(DATA['streamlines'], affine_to_rasmm=np.eye(4))
 
         # Just to create a generator
-        for ext, _ in FORMATS.items():
+        for ext in FORMATS:
             filtered = (s for s in tractogram.streamlines if True)
             lazy_tractogram = LazyTractogram(lambda: filtered, affine_to_rasmm=np.eye(4))
 

nibabel/testing/helpers.py

@@ -14,7 +14,7 @@
 def bytesio_filemap(klass):
     """Return bytes io filemap for this image class `klass`"""
     file_map = klass.make_file_map()
-    for name, fileholder in file_map.items():
+    for fileholder in file_map.values():
         fileholder.fileobj = BytesIO()
         fileholder.pos = 0
     return file_map