[Python-checkins] r78590 - python/trunk/Lib/test/test_argparse.py

[benjamin.peterson]

Author: benjamin.peterson
Date: Tue Mar  2 23:20:10 2010
New Revision: 78590

Log:
enable running of argparse tests and fix two that failed in the new environment

Modified:
   python/trunk/Lib/test/test_argparse.py

Modified: python/trunk/Lib/test/test_argparse.py
==============================================================================
--- python/trunk/Lib/test/test_argparse.py	(original)
+++ python/trunk/Lib/test/test_argparse.py	Tue Mar  2 23:20:10 2010
@@ -21,6 +21,8 @@
 import unittest
 import argparse
 
+from test import test_support
+
 try:
     from StringIO import StringIO
 except ImportError:
@@ -1951,6 +1953,8 @@
         group.add_argument('-a', action='store_true')
         group.add_argument('-b', action='store_true')
 
+        self.main_program = os.path.basename(sys.argv[0])
+
     def test_single_parent(self):
         parser = ErrorRaisingArgumentParser(parents=[self.wxyz_parent])
         self.assertEqual(parser.parse_args('-y 1 2 --w 3'.split()),
@@ -2043,7 +2047,7 @@
         parser = ErrorRaisingArgumentParser(parents=parents)
         parser_help = parser.format_help()
         self.assertEqual(parser_help, textwrap.dedent('''\
-            usage: test_argparse.py [-h] [-b B] [--d D] [--w W] [-y Y] a z
+            usage: {} [-h] [-b B] [--d D] [--w W] [-y Y] a z
 
             positional arguments:
               a
@@ -2059,7 +2063,7 @@
 
             x:
               -y Y
-        '''))
+        '''.format(self.main_program)))
 
     def test_groups_parents(self):
         parent = ErrorRaisingArgumentParser(add_help=False)
@@ -2076,7 +2080,7 @@
 
         parser_help = parser.format_help()
         self.assertEqual(parser_help, textwrap.dedent('''\
-            usage: test_argparse.py [-h] [-w W] [-x X] [-y Y | -z Z]
+            usage: {} [-h] [-w W] [-x X] [-y Y | -z Z]
 
             optional arguments:
               -h, --help  show this help message and exit
@@ -2088,7 +2092,7 @@
 
               -w W
               -x X
-        '''))
+        '''.format(self.main_program)))
 
 # ==============================
 # Mutually exclusive group tests
@@ -4199,6 +4203,9 @@
         for name in argparse.__all__:
             self.failUnless(hasattr(argparse, name))
 
+def test_main():
+    test_support.run_unittest(__name__)
+
 
 if __name__ == '__main__':
     unittest.main()