TEST: Added brikhead to test_image_api.py

Created minimal test_image_api test for brikhead, modeled on PARREC image tests.

Author: rmarkello

nibabel/brikhead.py

@@ -447,7 +447,7 @@ class AFNIImage(SpatialImage):
     """
 
     header_class = AFNIHeader
-    valid_exts = ('.BRIK', '.HEAD')  # HEAD, at least, needs to be upper
+    valid_exts = ('.brik', '.head')
     files_types = (('image', '.brik'), ('header', '.head'))
     _compressed_suffixes = ('.gz', '.bz2')
     makeable = False

nibabel/tests/test_brikhead.py

@@ -25,7 +25,7 @@
 EXAMPLE_IMAGES = [
     dict(
         head=pjoin(data_path, 'example4d+orig.HEAD'),
-        brik=pjoin(data_path, 'example4d+orig.BRIK.gz'),
+        fname=pjoin(data_path, 'example4d+orig.BRIK.gz'),
         shape=(33, 41, 25, 3),
         dtype=np.int16,
         affine=np.array([[-3.0,0,0,49.5],
@@ -43,7 +43,7 @@
         scaling=None),
     dict(
         head=pjoin(data_path, 'scaled+tlrc.HEAD'),
-        brik=pjoin(data_path, 'scaled+tlrc.BRIK'),
+        fname=pjoin(data_path, 'scaled+tlrc.BRIK'),
         shape=(47, 54, 43, 1.),
         dtype=np.int16,
         affine=np.array([[3.0,0,0,-66.],
@@ -85,7 +85,7 @@ def test_makehead(self):
             with assert_raises(self.module.AFNIHeaderError):
                 self.module.AFNIHeader.from_header(header=None)
             with assert_raises(self.module.AFNIHeaderError):
-                self.module.AFNIHeader.from_header(tp['brik'])
+                self.module.AFNIHeader.from_header(tp['fname'])
 
 
 class TestAFNIImage(object):
@@ -94,7 +94,7 @@ class TestAFNIImage(object):
 
     def test_brikheadfile(self):
         for tp in self.test_files:
-            brik = self.module.load(tp['brik'])
+            brik = self.module.load(tp['fname'])
             assert_equal(brik.get_data_dtype().type, tp['dtype'])
             assert_equal(brik.shape, tp['shape'])
             assert_equal(brik.header.get_zooms(), tp['zooms'])
@@ -120,7 +120,7 @@ def test_load(self):
     def test_array_proxy_slicing(self):
         # Test slicing of array proxy
         for tp in self.test_files:
-            img = self.module.load(tp['brik'])
+            img = self.module.load(tp['fname'])
             arr = img.get_data()
             prox = img.dataobj
             assert_true(prox.is_proxy)

nibabel/tests/test_image_api.py

@@ -39,7 +39,7 @@
                 Nifti1Pair, Nifti1Image, Nifti2Pair, Nifti2Image,
                 MGHImage, Minc1Image, Minc2Image, is_proxy)
 from ..spatialimages import SpatialImage
-from .. import minc1, minc2, parrec
+from .. import minc1, minc2, parrec, brikhead
 
 from nose import SkipTest
 from nose.tools import (assert_true, assert_false, assert_raises, assert_equal)
@@ -54,7 +54,7 @@
 from .test_minc1 import EXAMPLE_IMAGES as MINC1_EXAMPLE_IMAGES
 from .test_minc2 import EXAMPLE_IMAGES as MINC2_EXAMPLE_IMAGES
 from .test_parrec import EXAMPLE_IMAGES as PARREC_EXAMPLE_IMAGES
-
+from .test_brikhead import EXAMPLE_IMAGES as AFNI_EXAMPLE_IMAGES
 
 class GenericImageAPI(ValidateAPI):
     """ General image validation API """
@@ -596,3 +596,9 @@ class TestMGHAPI(ImageHeaderAPI):
     has_scaling = True
     can_save = True
     standard_extension = '.mgh'
+
+
+class TestAFNIAPI(LoadImageAPI):
+    loader = brikhead.load
+    klass = image_maker = brikhead.AFNIImage
+    example_images = AFNI_EXAMPLE_IMAGES