TEST: Load CaretSpecFile as a GeometryCollection

Author: effigies

nibabel/coordimage.py

@@ -107,6 +107,9 @@ class GeometryCollection:
         Mapping from structure names to ``Pointset``
     """
 
+    def __init__(self, structures):
+        self.structures = structures
+
     @classmethod
     def from_spec(klass, pathlike):
         """Load a collection of geometries from a specification."""

nibabel/tests/test_coordimage.py

@@ -28,11 +28,27 @@ def from_spec(klass, pathlike):
             'lh': FreeSurferHemisphere.from_filename(surfs / 'lh.white'),
             'rh': FreeSurferHemisphere.from_filename(surfs / 'rh.white'),
         }
-        subject = super().__init__(structures)
+        subject = klass(structures)
         subject._subject_dir = subject_dir
         return subject
 
 
+class CaretSpec(ci.GeometryCollection):
+    @classmethod
+    def from_spec(klass, pathlike):
+        from nibabel.cifti2.caretspec import CaretSpecFile
+
+        csf = CaretSpecFile.from_filename(pathlike)
+        structures = {
+            df.structure: df.uri
+            for df in csf.data_files
+            if df.selected  # Use selected to avoid overloading for now
+        }
+        wbspec = klass(structures)
+        wbspec._specfile = csf
+        return wbspec
+
+
 def test_Cifti2Image_as_CoordImage():
     ones = nb.load(CIFTI2_DATA / 'ones.dscalar.nii')
     axes = [ones.header.get_axis(i) for i in range(ones.ndim)]