Skip to content

Commit 959a1c7

Browse files
committed
RF: use generic testing function to hdr1=hdr2
Use ``assert_arr_dict_equal`` to test whether two headers are equal, rather than custom testing function.
1 parent bf8b6b5 commit 959a1c7

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

nibabel/streamlines/tests/test_trk.py

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -82,14 +82,6 @@ def setup():
8282
affine_to_rasmm=np.eye(4))
8383

8484

85-
def assert_header_equal(h1, h2):
86-
for k in h1.keys():
87-
assert_array_equal(h2[k], h1[k])
88-
89-
for k in h2.keys():
90-
assert_array_equal(h1[k], h2[k])
91-
92-
9385
class TestTRK(unittest.TestCase):
9486

9587
def test_load_empty_file(self):
@@ -298,7 +290,7 @@ def test_load_write_LPS_file(self):
298290

299291
new_trk = TrkFile.load(trk_file)
300292

301-
assert_header_equal(new_trk.header, trk.header)
293+
assert_arr_dict_equal(new_trk.header, trk.header)
302294
assert_tractogram_equal(new_trk.tractogram, trk.tractogram)
303295

304296
new_trk_orig = TrkFile.load(DATA['standard_LPS_trk_fname'])
@@ -322,7 +314,7 @@ def test_load_write_LPS_file(self):
322314

323315
new_trk = TrkFile.load(trk_file)
324316

325-
assert_header_equal(new_trk.header, trk_LPS.header)
317+
assert_arr_dict_equal(new_trk.header, trk_LPS.header)
326318
assert_tractogram_equal(new_trk.tractogram, trk.tractogram)
327319

328320
new_trk_orig = TrkFile.load(DATA['standard_LPS_trk_fname'])

0 commit comments

Comments
 (0)