Use importlib instead of calling __import__ directly.

This commit is contained in:
Brett Cannon 2012-11-14 13:42:51 -05:00
parent 2eb2f5e359
commit 613cf25d2a
1 changed files with 1 additions and 2 deletions

View File

@ -1210,8 +1210,7 @@ def runtest_inner(test, verbose, quiet,
abstest = 'test.' + test abstest = 'test.' + test
with saved_test_environment(test, verbose, quiet) as environment: with saved_test_environment(test, verbose, quiet) as environment:
start_time = time.time() start_time = time.time()
the_package = __import__(abstest, globals(), locals(), []) the_module = importlib.import_module(abstest)
the_module = getattr(the_package, test)
# If the test has a test_main, that will run the appropriate # If the test has a test_main, that will run the appropriate
# tests. If not, use normal unittest test loading. # tests. If not, use normal unittest test loading.
test_runner = getattr(the_module, "test_main", None) test_runner = getattr(the_module, "test_main", None)