Skip to content

Commit aeb7a8d

Browse files
STY: Apply ruff/Perflint rule PERF102
PERF102 When using only the keys of a dict use the `keys()` method PERF102 When using only the values of a dict use the `values()` method
1 parent c742552 commit aeb7a8d

File tree

6 files changed

+14
-14
lines changed

6 files changed

+14
-14
lines changed

nibabel/streamlines/tests/test_streamlines.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ def test_save_tractogram_file(self):
207207

208208
def test_save_empty_file(self):
209209
tractogram = Tractogram(affine_to_rasmm=np.eye(4))
210-
for ext, cls in FORMATS.items():
210+
for ext in FORMATS:
211211
with InTemporaryDirectory():
212212
filename = 'streamlines' + ext
213213
nib.streamlines.save(tractogram, filename)
@@ -216,7 +216,7 @@ def test_save_empty_file(self):
216216

217217
def test_save_simple_file(self):
218218
tractogram = Tractogram(DATA['streamlines'], affine_to_rasmm=np.eye(4))
219-
for ext, cls in FORMATS.items():
219+
for ext in FORMATS:
220220
with InTemporaryDirectory():
221221
filename = 'streamlines' + ext
222222
nib.streamlines.save(tractogram, filename)
@@ -262,7 +262,7 @@ def test_save_complex_file(self):
262262
def test_save_sliced_tractogram(self):
263263
tractogram = Tractogram(DATA['streamlines'], affine_to_rasmm=np.eye(4))
264264
original_tractogram = tractogram.copy()
265-
for ext, cls in FORMATS.items():
265+
for ext in FORMATS:
266266
with InTemporaryDirectory():
267267
filename = 'streamlines' + ext
268268
nib.streamlines.save(tractogram[::2], filename)
@@ -283,7 +283,7 @@ def test_save_from_generator(self):
283283
tractogram = Tractogram(DATA['streamlines'], affine_to_rasmm=np.eye(4))
284284

285285
# Just to create a generator
286-
for ext, _ in FORMATS.items():
286+
for ext in FORMATS:
287287
filtered = (s for s in tractogram.streamlines if True)
288288
lazy_tractogram = LazyTractogram(lambda: filtered, affine_to_rasmm=np.eye(4))
289289

nibabel/testing/helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
def bytesio_filemap(klass):
1515
"""Return bytes io filemap for this image class `klass`"""
1616
file_map = klass.make_file_map()
17-
for name, fileholder in file_map.items():
17+
for fileholder in file_map.values():
1818
fileholder.fileobj = BytesIO()
1919
fileholder.pos = 0
2020
return file_map

nibabel/tests/test_analyze.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -728,7 +728,7 @@ def test_data_hdr_cache(self):
728728
IC = self.image_class
729729
# save an image to a file map
730730
fm = IC.make_file_map()
731-
for key, value in fm.items():
731+
for key in fm:
732732
fm[key].fileobj = BytesIO()
733733
shape = (2, 3, 4)
734734
data = np.arange(24, dtype=np.int8).reshape(shape)
@@ -831,7 +831,7 @@ def test_header_updating(self):
831831
hdr = img.header
832832
hdr.set_zooms((4, 5, 6))
833833
# Save / reload using bytes IO objects
834-
for key, value in img.file_map.items():
834+
for value in img.file_map.values():
835835
value.fileobj = BytesIO()
836836
img.to_file_map()
837837
hdr_back = img.from_file_map(img.file_map).header
@@ -842,7 +842,7 @@ def test_header_updating(self):
842842
assert_array_equal(hdr.get_zooms(), (2, 3, 4))
843843
# Modify affine in-place? Update on save.
844844
img.affine[0, 0] = 9
845-
for key, value in img.file_map.items():
845+
for value in img.file_map.values():
846846
value.fileobj = BytesIO()
847847
img.to_file_map()
848848
hdr_back = img.from_file_map(img.file_map).header
@@ -864,7 +864,7 @@ def test_pickle(self):
864864
assert_array_equal(img.get_fdata(), img2.get_fdata())
865865
assert img.header == img2.header
866866
# Save / reload using bytes IO objects
867-
for key, value in img.file_map.items():
867+
for value in img.file_map.values():
868868
value.fileobj = BytesIO()
869869
img.to_file_map()
870870
img_prox = img.from_file_map(img.file_map)

nibabel/tests/test_files_interface.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ def test_files_spatialimages():
2828
]
2929
for klass in klasses:
3030
file_map = klass.make_file_map()
31-
for key, value in file_map.items():
31+
for value in file_map.values():
3232
assert value.filename is None
3333
assert value.fileobj is None
3434
assert value.pos == 0
@@ -41,7 +41,7 @@ def test_files_spatialimages():
4141
img = klass(arr.astype(np.float32), aff)
4242
else:
4343
img = klass(arr, aff)
44-
for key, value in img.file_map.items():
44+
for value in img.file_map.values():
4545
assert value.filename is None
4646
assert value.fileobj is None
4747
assert value.pos == 0

nibabel/tests/test_nifti1.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -820,7 +820,7 @@ def _qform_rt(self, img):
820820
hdr['qform_code'] = 3
821821
hdr['sform_code'] = 4
822822
# Save / reload using bytes IO objects
823-
for key, value in img.file_map.items():
823+
for value in img.file_map.values():
824824
value.fileobj = BytesIO()
825825
img.to_file_map()
826826
return img.from_file_map(img.file_map)

nibabel/tests/test_spm99analyze.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,7 @@ def test_mat_read(self):
423423
aff = np.diag([2, 3, 4, 1]) # no LR flip in affine
424424
img = img_klass(arr, aff)
425425
fm = img.file_map
426-
for key, value in fm.items():
426+
for value in fm.values():
427427
value.fileobj = BytesIO()
428428
# Test round trip
429429
img.to_file_map()
@@ -475,7 +475,7 @@ def test_none_affine(self):
475475
img = img_klass(np.zeros((2, 3, 4)), None)
476476
aff = img.header.get_best_affine()
477477
# Save / reload using bytes IO objects
478-
for key, value in img.file_map.items():
478+
for value in img.file_map.values():
479479
value.fileobj = BytesIO()
480480
img.to_file_map()
481481
img_back = img.from_file_map(img.file_map)

0 commit comments

Comments
 (0)