Browse Source

script: added flag `--version`

Fabian Peter Hammerle 5 years ago
parent
commit
06250eee7c
2 changed files with 11 additions and 2 deletions
  1. 2 1
      freesurfer_volume_reader/__main__.py
  2. 9 1
      tests/main_test.py

+ 2 - 1
freesurfer_volume_reader/__main__.py

@@ -10,7 +10,7 @@ import typing
 
 import pandas
 
-from freesurfer_volume_reader import ashs, freesurfer, parse_version_string, \
+from freesurfer_volume_reader import __version__, ashs, freesurfer, parse_version_string, \
                                      remove_group_names_from_regex
 
 def concat_dataframes(dataframes: typing.Iterable[pandas.DataFrame]
@@ -48,6 +48,7 @@ def main():
                            nargs='*' if subjects_dir_path else '+',
                            default=[subjects_dir_path],
                            help='default: $SUBJECTS_DIR ({})'.format(subjects_dir_path))
+    argparser.add_argument('--version', action='version', version=__version__)
     args = argparser.parse_args()
     filename_regexs = {k[len('filename_regex.'):]: v for k, v in vars(args).items()
                        if k.startswith('filename_regex.')}

+ 9 - 1
tests/main_test.py

@@ -8,6 +8,7 @@ import pandas
 import pandas.util.testing
 import pytest
 
+import freesurfer_volume_reader
 import freesurfer_volume_reader.__main__
 
 from conftest import SUBJECTS_DIR, assert_volume_frames_equal
@@ -201,6 +202,13 @@ def test_main_root_dir_filename_regex_combined(capsys):
     )
 
 
-def test_main_module_script():
+def test_main_module_script_help():
     subprocess.run(['python', '-m', 'freesurfer_volume_reader', '--help'],
                    check=True)
+
+
+def test_main_module_script_version():
+    proc_info = subprocess.run(['python', '-m', 'freesurfer_volume_reader', '--version'],
+                               check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    assert proc_info.stdout.rstrip() == freesurfer_volume_reader.__version__.encode()
+    assert not proc_info.stderr