Skip to content

Commit a11cd20

Browse files
committed
TEST: Make xform code test more concise
1 parent 288bbf7 commit a11cd20

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

nibabel/tests/test_nifti1.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -229,14 +229,12 @@ def test_nifti_qsform_checks(self):
229229
def test_nifti_xform_codes(self):
230230
# Verify that all xform codes can be set in both qform and sform
231231
hdr = self.header_class()
232-
xform_codes = nifti1.xform_codes
233-
all_codes = list(xform_codes.keys())
234232
affine = np.eye(4)
235-
for code in all_codes:
233+
for code in nifti1.xform_codes.keys():
236234
hdr.set_qform(affine, code)
237-
assert_equal(hdr['qform_code'], xform_codes[code])
235+
assert_equal(hdr['qform_code'], nifti1.xform_codes[code])
238236
hdr.set_sform(affine, code)
239-
assert_equal(hdr['sform_code'], xform_codes[code])
237+
assert_equal(hdr['sform_code'], nifti1.xform_codes[code])
240238

241239
def test_magic_offset_checks(self):
242240
# magic and offset

0 commit comments

Comments
 (0)