Make --help output clean by not running self-test.

This commit is contained in:
Nathan Wilcox 2017-05-19 13:09:24 -07:00
parent 1ecb494896
commit b43194baee
1 changed files with 14 additions and 11 deletions

View File

@ -251,16 +251,19 @@ class TestVersion (unittest.TestCase):
if __name__ == '__main__':
actualargs = sys.argv
sys.argv = [sys.argv[0], '--verbose']
if len(sys.argv) == 2 and sys.argv[1] == '--help':
main()
else:
actualargs = sys.argv
sys.argv = [sys.argv[0], '--verbose']
print '=== Self Test ==='
try:
unittest.main()
except SystemExit as e:
if e.args[0] != 0:
raise
print '=== Self Test ==='
try:
unittest.main()
except SystemExit as e:
if e.args[0] != 0:
raise
sys.argv = actualargs
print '=== Running ==='
main()
sys.argv = actualargs
print '=== Running ==='
main()