Skip to content

Commit db34949

Browse files
committed
tst: cover some edge cases
1 parent f8650e4 commit db34949

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

nitransforms/tests/test_nonlinear.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,11 +59,19 @@ def test_itk_disp_load_intent():
5959

6060

6161
def test_displacements_init():
62-
DenseFieldTransform(
62+
identity1 = DenseFieldTransform(
6363
np.zeros((10, 10, 10, 3)),
6464
reference=nb.Nifti1Image(np.zeros((10, 10, 10, 3)), np.eye(4), None),
6565
)
66+
identity2 = DenseFieldTransform(
67+
reference=nb.Nifti1Image(np.zeros((10, 10, 10)), np.eye(4), None),
68+
)
69+
70+
assert np.array_equal(identity1._field, identity2._field)
71+
assert np.array_equal(identity1._displacements, identity2._displacements)
6672

73+
with pytest.raises(TransformError):
74+
DenseFieldTransform()
6775
with pytest.raises(TransformError):
6876
DenseFieldTransform(np.zeros((10, 10, 10, 3)))
6977
with pytest.raises(TransformError):

0 commit comments

Comments
 (0)