TEST: Load CaretSpecFile as a GeometryCollection

Author: effigies

nibabel/coordimage.py

@@ -103,6 +103,8 @@ class GeometryCollection:
     structures : dict
         Mapping from structure names to ``Pointset``
     """
+    def __init__(self, structures):
+        self.structures = structures
 
     @classmethod
     def from_spec(klass, pathlike):

nibabel/tests/test_coordimage.py

@@ -27,11 +27,26 @@ 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)]