Skip to content
Snippets Groups Projects
Commit 9b9b4caf authored by Paul McCarthy's avatar Paul McCarthy :mountain_bicyclist:
Browse files

TEST: Update convert_x5 tests

parent 54f07a26
No related branches found
No related tags found
No related merge requests found
...@@ -60,7 +60,7 @@ def test_convert_flirt(): ...@@ -60,7 +60,7 @@ def test_convert_flirt():
assert np.all(np.isclose(gotxform, xform)) assert np.all(np.isclose(gotxform, xform))
def test_convert_fnirt_displacement_field(): def test_convert_fnirt_deformation_field():
datadir = op.join(op.dirname(__file__), '..', datadir = op.join(op.dirname(__file__), '..',
'test_transform', 'testdata', 'nonlinear') 'test_transform', 'testdata', 'nonlinear')
...@@ -86,7 +86,7 @@ def test_convert_fnirt_displacement_field(): ...@@ -86,7 +86,7 @@ def test_convert_fnirt_displacement_field():
assert dfnii.ref.sameSpace(ref) assert dfnii.ref.sameSpace(ref)
assert dfnii.srcSpace == df.srcSpace assert dfnii.srcSpace == df.srcSpace
assert dfnii.refSpace == df.refSpace assert dfnii.refSpace == df.refSpace
assert dfnii.displacementType == df.displacementType assert dfnii.deformationType == df.deformationType
assert np.all(np.isclose(dfnii.data, df.data)) assert np.all(np.isclose(dfnii.data, df.data))
...@@ -97,6 +97,7 @@ def test_convert_fnirt_coefficient_field(): ...@@ -97,6 +97,7 @@ def test_convert_fnirt_coefficient_field():
srcfile = op.join(datadir, 'src.nii.gz') srcfile = op.join(datadir, 'src.nii.gz')
reffile = op.join(datadir, 'ref.nii.gz') reffile = op.join(datadir, 'ref.nii.gz')
cffile = op.join(datadir, 'coefficientfield.nii.gz') cffile = op.join(datadir, 'coefficientfield.nii.gz')
dffile = op.join(datadir, 'displacementfield.nii.gz')
with tempdir.tempdir(): with tempdir.tempdir():
...@@ -109,15 +110,17 @@ def test_convert_fnirt_coefficient_field(): ...@@ -109,15 +110,17 @@ def test_convert_fnirt_coefficient_field():
src = fslimage.Image(srcfile) src = fslimage.Image(srcfile)
ref = fslimage.Image(reffile) ref = fslimage.Image(reffile)
cf = fnirt.readFnirt(cffile, src, ref) df = fnirt.readFnirt(dffile, src, ref)
cfnii = fnirt.readFnirt('coef.nii.gz', src, ref) dfnii = fnirt.readFnirt('coef.nii.gz', src, ref)
assert cfnii.src.sameSpace(src) assert dfnii .sameSpace(df)
assert cfnii.ref.sameSpace(ref) assert dfnii.src.sameSpace(src)
assert cfnii.srcSpace == cf.srcSpace assert dfnii.ref.sameSpace(ref)
assert cfnii.refSpace == cf.refSpace
assert cfnii.knotSpacing == cf.knotSpacing assert dfnii.srcSpace == df.srcSpace
assert dfnii.refSpace == df.refSpace
assert np.all(np.isclose(cfnii.fieldToRefMat, cf.fieldToRefMat)) assert dfnii.deformationType == 'relative'
assert np.all(np.isclose(cfnii.srcToRefMat, cf.srcToRefMat))
assert np.all(np.isclose(cfnii.data, cf.data)) diff = np.abs(dfnii.data - df.data)
tols = {'rtol' : 1e-5, 'atol' : 1e-5}
assert np.all(np.isclose(dfnii.data, df.data, **tols))
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment