Merge pull request #33 from int-brain-lab/ucla

iter session returns the session path if a session path is provided

one/alf/io.py

@@ -366,6 +366,8 @@ def iter_sessions(root_dir):
     pathlib.Path
         The next session path in lexicographical order
     """
+    if spec.is_session_path(root_dir):
+        yield root_dir
     for path in sorted(Path(root_dir).rglob('*')):
         if path.is_dir() and spec.is_session_path(path):
             yield path

one/tests/alf/test_alf_io.py

@@ -584,6 +584,8 @@ def test_iter_sessions(self):
         valid_sessions = alfio.iter_sessions(self.tempdir.name)
         self.assertEqual(next(valid_sessions), self.session_path)
         self.assertFalse(next(valid_sessions, False))
+        # makes sure that the session path returns itself on the iterator
+        self.assertEqual(self.session_path, next(alfio.iter_sessions(self.session_path)))
 
 
 if __name__ == "__main__":

setup.py

@@ -24,7 +24,7 @@
 
 setup(
     name='ONE-api',
-    version='1.7.2',
+    version='1.7.3',
     python_requires='>={}.{}'.format(*REQUIRED_PYTHON),
     description='Open Neurophysiology Environment',
     license="MIT",