TST: Add simple test for label file parsing

Author: mgxd

nibabies/interfaces/conftest.py renamed to nibabies/conftest.py

@@ -1,8 +1,9 @@
 """py.test configuration"""
 from pathlib import Path
-import pytest
 from tempfile import TemporaryDirectory
+from pkg_resources import resource_filename
 
+import pytest
 
 FILES = (
     'functional.nii',
@@ -32,3 +33,4 @@ def data_dir():
 @pytest.fixture(autouse=True)
 def set_namespace(doctest_namespace, data_dir):
     doctest_namespace["data_dir"] = data_dir
+    doctest_namespace["test_data"] = Path(resource_filename("nibabies", "tests/data"))

nibabies/tests/data/labelfile.txt

@@ -0,0 +1,6 @@
+CEREBELLUM_LEFT
+8 230 148 34 255
+THALAMUS_LEFT
+10 0 118 14 255
+CAUDATE_LEFT
+11 122 186 220 255

nibabies/workflows/bold/alignment.py

@@ -232,14 +232,11 @@ def parse_roi_labels(label_file):
 
     Return a list of structure names and label keys.
 
-    Example
-    -------
-    CEREBELLUM_LEFT
-    8 230 148 34 255
-    THALAMUS_LEFT
-    10 0 118 14 255
-
-    TODO: Add unit test
+    >>> structs, ids = parse_roi_labels(test_data / "labelfile.txt")
+    >>> structs
+    '["CEREBELLUM_LEFT", "THALAMUS_LEFT", "CAUDATE_LEFT"]'
+    >>> ids
+    '[8, 10, 11]'
     """
 
     with open(label_file) as fp: