From 15bbdc375f8a9f7c2937ec32207e534ba432fac7 Mon Sep 17 00:00:00 2001 From: Yuya Unno Date: Fri, 12 May 2017 16:17:05 +0900 Subject: [PATCH 1/3] Add verison option --- cudnnenv/__init__.py | 2 +- test/test_command.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/cudnnenv/__init__.py b/cudnnenv/__init__.py index f8ca172..276ab34 100644 --- a/cudnnenv/__init__.py +++ b/cudnnenv/__init__.py @@ -348,7 +348,7 @@ def deactivate(args): def main(args=None): - parser = argparse.ArgumentParser() + parser = argparse.ArgumentParser(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 index 978f0b7..39e2ee4 100644 --- a/test/test_command.py +++ b/test/test_command.py @@ -48,6 +48,9 @@ def clear_stdout(self): self.stdout = StringIO() sys.stdout = self.stdout + def test_version(self): + self.call_main('--version') + def test_no_subcommand(self): with self.assertRaises(SystemExit) as cont: self.call_main() From 2473cf916831b89c956b2a1d10c5247de9059443 Mon Sep 17 00:00:00 2001 From: Yuya Unno Date: Sun, 14 May 2017 21:38:07 +0900 Subject: [PATCH 2/3] Insert assertRaises --- test/test_command.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/test_command.py b/test/test_command.py index 39e2ee4..46f65d4 100644 --- a/test/test_command.py +++ b/test/test_command.py @@ -49,7 +49,8 @@ def clear_stdout(self): sys.stdout = self.stdout def test_version(self): - self.call_main('--version') + with self.assertRaises(SystemExit) as cont: + self.call_main('--version') def test_no_subcommand(self): with self.assertRaises(SystemExit) as cont: From 341f4b6e2ae920045c34cedd95b0ad609c22691d Mon Sep 17 00:00:00 2001 From: Yuya Unno Date: Mon, 15 May 2017 01:12:56 +0900 Subject: [PATCH 3/3] Use version action --- cudnnenv/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cudnnenv/__init__.py b/cudnnenv/__init__.py index 276ab34..fdcc93b 100644 --- a/cudnnenv/__init__.py +++ b/cudnnenv/__init__.py @@ -348,7 +348,9 @@ def deactivate(args): def main(args=None): - parser = argparse.ArgumentParser(version='cudnnenv %s' % __version__) + 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')