mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-11-04 11:49:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 11

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

Browse source
This commit is contained in:
Brett Cannon 2012-02-17 09:37:39 -05:00
parent ba17fe256e
commit 1f14bebe3c
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
Lib/importlib/test/__main__.py
View file

@ -4,19 +4,24 @@ Specifying the ``--builtin`` flag will run tests, where applicable, with
builtins.__import__ instead of importlib.__import__. builtins.__import__ instead of importlib.__import__.
""" """
import argparse
from importlib.test.import_ import util from importlib.test.import_ import util
import os.path import os.path
from test.support import run_unittest from test.support import run_unittest
import sys
import unittest import unittest
def test_main(): 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__) start_dir = os.path.dirname(__file__)
top_dir = os.path.dirname(os.path.dirname(start_dir)) top_dir = os.path.dirname(os.path.dirname(start_dir))
test_loader = unittest.TestLoader() 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)) run_unittest(test_loader.discover(start_dir, top_level_dir=top_dir))