Skip to content

Commit 4740545

Browse files
committed
added test for loading .trk with lowercase voxel order in hdr
1 parent 419998a commit 4740545

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

nibabel/streamlines/tests/test_trk.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
from ..tractogram_file import HeaderError, HeaderWarning
1818

1919
from .. import trk as trk_module
20-
from ..trk import TrkFile, encode_value_in_name, decode_value_from_name
20+
from ..trk import TrkFile, encode_value_in_name, decode_value_from_name, get_affine_trackvis_to_rasmm
2121
from ..header import Field
2222

2323
DATA = {}
@@ -110,6 +110,17 @@ def trk_with_bytes(self, trk_key='simple_trk_fname', endian='<'):
110110
return trk_struct, trk_bytes
111111

112112
def test_load_file_with_wrong_information(self):
113+
# Simulate a TRK file where `voxel_order` is lowercase.
114+
trk_struct1, trk_bytes1 = self.trk_with_bytes()
115+
trk_struct1[Field.VOXEL_ORDER] = b'LAS'
116+
trk1 = TrkFile.load(BytesIO(trk_bytes1))
117+
trk_struct2, trk_bytes2 = self.trk_with_bytes()
118+
trk_struct2[Field.VOXEL_ORDER] = b'las'
119+
trk2 = TrkFile.load(BytesIO(trk_bytes2))
120+
trk1_aff2rasmm = get_affine_trackvis_to_rasmm(trk1.header)
121+
trk2_aff2rasmm = get_affine_trackvis_to_rasmm(trk2.header)
122+
assert_array_equal(trk1_aff2rasmm,trk2_aff2rasmm)
123+
113124
# Simulate a TRK file where `count` was not provided.
114125
trk_struct, trk_bytes = self.trk_with_bytes()
115126
trk_struct[Field.NB_STREAMLINES] = 0

0 commit comments

Comments
 (0)