NF tests for EcatImage

Author: cindeem

nibabel/ecat.py

@@ -15,7 +15,7 @@
 from nibabel.volumeutils import make_dt_codes, allopen
 
 
-from nibabel.spatialimages import SpatialImage, HeaderDataError, HeaderTypeError
+from nibabel.spatialimages import SpatialImage, HeaderDataError, HeaderTypeError, ImageDataError
 from nibabel.volumeutils import allopen, array_from_file
 from .fileholders import FileHolderError, copy_file_map
 from .arrayproxy import ArrayProxy

nibabel/tests/test_ecat.py

@@ -118,3 +118,26 @@ def test_subheader(self):
         scaled_dat = self.subhdr.data_from_fileobj()
         yield assert_array_equal(dat * scale_factor * ecat_calib_factor,
                                  scaled_dat)
+
+
+class TestEcatImage(ParametricTestCase):
+    image_class = EcatImage
+    example_file = ecat_file
+    img = image_class.load(example_file)
+
+    def test_file(self):
+        yield assert_equal(self.img.file_map['header'].filename,
+                           self.example_file)
+        yield assert_equal(self.img.file_map['image'].filename,
+                           self.example_file)
+        yield assert_raises(NotImplementedError,
+                            self.img.to_filename,
+                            'tmp.v')
+
+    def test_data(self):
+        dat = self.img.get_data()
+        yield assert_equal(dat.shape, self.img.get_shape())
+        frame = self.img.get_frame(0)
+        yield assert_array_equal(frame, dat[:,:,:,0])
+        
+