Merge pull request #46 from unnonouno/version-option

Add verison option
unnonouno · May 15, 2017 · cb1ba1f · cb1ba1f
2 parents fd6bd1a + 341f4b6
commit cb1ba1f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/cudnnenv/__init__.py b/cudnnenv/__init__.py
@@ -349,6 +349,8 @@ def deactivate(args):
 
 def main(args=None):
     parser = argparse.ArgumentParser()
+    parser.add_argument(
+        '--version', action='version', version='cudnnenv %s' % __version__)
     subparsers = parser.add_subparsers(help='Subcommand')
 
     sub = subparsers.add_parser('install', help='Install version')

diff --git a/test/test_command.py b/test/test_command.py
@@ -48,6 +48,10 @@ def clear_stdout(self):
         self.stdout = StringIO()
         sys.stdout = self.stdout
 
+    def test_version(self):
+        with self.assertRaises(SystemExit) as cont:
+            self.call_main('--version')
+
     def test_no_subcommand(self):
         with self.assertRaises(SystemExit) as cont:
             self.call_main()