From 5196449a26e08f2214285c2dd99ad906be361630 Mon Sep 17 00:00:00 2001
From: Paul McCarthy <pauldmccarthy@gmail.com>
Date: Wed, 28 Feb 2018 20:44:33 +0000
Subject: [PATCH] adjust unit test for ensureIsImage
---
tests/test_ensure.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/tests/test_ensure.py b/tests/test_ensure.py
index dabd3a952..865c205bb 100644
--- a/tests/test_ensure.py
+++ b/tests/test_ensure.py
@@ -22,7 +22,9 @@ def test_ensureIsImage():
assert ensure.ensureIsImage(img) is img
- loaded = ensure.ensureIsImage('image.nii')
+ loaded = [ensure.ensureIsImage('image.nii'),
+ ensure.ensureIsImage('image')]
- assert isinstance(loaded, nib.nifti1.Nifti1Image)
- assert np.all(img.get_data() == loaded.get_data())
+ for l in loaded:
+ assert isinstance(l, nib.nifti1.Nifti1Image)
+ assert np.all(img.get_data() == l.get_data())
--
GitLab