File tree Expand file tree Collapse file tree 1 file changed +2
-10
lines changed
nibabel/streamlines/tests Expand file tree Collapse file tree 1 file changed +2
-10
lines changed Original file line number Diff line number Diff line change @@ -82,14 +82,6 @@ def setup():
82
82
affine_to_rasmm = np .eye (4 ))
83
83
84
84
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
-
93
85
class TestTRK (unittest .TestCase ):
94
86
95
87
def test_load_empty_file (self ):
@@ -298,7 +290,7 @@ def test_load_write_LPS_file(self):
298
290
299
291
new_trk = TrkFile .load (trk_file )
300
292
301
- assert_header_equal (new_trk .header , trk .header )
293
+ assert_arr_dict_equal (new_trk .header , trk .header )
302
294
assert_tractogram_equal (new_trk .tractogram , trk .tractogram )
303
295
304
296
new_trk_orig = TrkFile .load (DATA ['standard_LPS_trk_fname' ])
@@ -322,7 +314,7 @@ def test_load_write_LPS_file(self):
322
314
323
315
new_trk = TrkFile .load (trk_file )
324
316
325
- assert_header_equal (new_trk .header , trk_LPS .header )
317
+ assert_arr_dict_equal (new_trk .header , trk_LPS .header )
326
318
assert_tractogram_equal (new_trk .tractogram , trk .tractogram )
327
319
328
320
new_trk_orig = TrkFile .load (DATA ['standard_LPS_trk_fname' ])
You can’t perform that action at this time.
0 commit comments