From: Brett Cannon <brett@python.org>
Date: Fri, 17 Feb 2012 14:37:39 +0000 (-0500)
Subject: Have importlib.test use argparse instead of some hacked up solution.
X-Git-Tag: v3.3.0a1~156
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=1f14bebe3c8e13957867c9f0aa55feb9e767cd0a;p=thirdparty%2FPython%2Fcpython.git

Have importlib.test use argparse instead of some hacked up solution.
---

diff --git a/Lib/importlib/test/__main__.py b/Lib/importlib/test/__main__.py
index a1990b1f0177..55158126da55 100644
--- a/Lib/importlib/test/__main__.py
+++ b/Lib/importlib/test/__main__.py
@@ -4,19 +4,24 @@ Specifying the ``--builtin`` flag will run tests, where applicable, with
 builtins.__import__ instead of importlib.__import__.
 
 """
+import argparse
 from importlib.test.import_ import util
 import os.path
 from test.support import run_unittest
-import sys
 import unittest
 
 
 def test_main():
+    parser = argparse.ArgumentParser(description='Execute the importlib test '
+                                                  'suite')
+    parser.add_argument('-b', '--builtin', action='store_true', default=False,
+                        help='use builtins.__import__() instead of importlib')
+    args = parser.parse_args()
+    if args.builtin:
+        util.using___import__ = True
     start_dir = os.path.dirname(__file__)
     top_dir = os.path.dirname(os.path.dirname(start_dir))
     test_loader = unittest.TestLoader()
-    if '--builtin' in sys.argv:
-        util.using___import__ = True
     run_unittest(test_loader.discover(start_dir, top_level_dir=top_dir))