unit tests for ensure module

tests/test_ensure.py

+#!/usr/bin/env python
+#
+# test_ensure.py -
+#
+# Author: Paul McCarthy <pauldmccarthy@gmail.com>
+#
+
+
+import numpy   as np
+import nibabel as nib
+
+import fsl.utils.tempdir as tempdir
+import fsl.utils.ensure  as ensure
+
+from . import make_random_image
+
+
+def test_ensureIsImage():
+
+    with tempdir.tempdir():
+        img = make_random_image('image.nii')
+
+        assert ensure.ensureIsImage(img) is img
+
+        loaded = ensure.ensureIsImage('image.nii')
+
+        assert isinstance(loaded, nib.nifti1.Nifti1Image)
+        assert np.all(img.get_data() == loaded.get_data())