[Python-checkins] cpython: Issue #14585: test_import now runs all tests under
brett.cannon
python-checkins at python.org
Fri Apr 20 21:52:22 CEST 2012
http://hg.python.org/cpython/rev/a74ba7407457
changeset: 76437:a74ba7407457
user: Brett Cannon <brett at python.org>
date: Fri Apr 20 15:52:17 2012 -0400
summary:
Issue #14585: test_import now runs all tests under
importlib.test.import_ using builtins.__import__() instead of just the
relative import tests.
files:
Lib/importlib/test/__init__.py | 25 ++++++++++++++++++++++
Lib/test/test_import.py | 25 ++++++++-------------
2 files changed, 35 insertions(+), 15 deletions(-)
diff --git a/Lib/importlib/test/__init__.py b/Lib/importlib/test/__init__.py
--- a/Lib/importlib/test/__init__.py
+++ b/Lib/importlib/test/__init__.py
@@ -0,0 +1,25 @@
+import os
+import sys
+import unittest
+
+def test_suite(package=__package__, directory=os.path.dirname(__file__)):
+ suite = unittest.TestSuite()
+ for name in os.listdir(directory):
+ if name.startswith(('.', '__')):
+ continue
+ path = os.path.join(directory, name)
+ if (os.path.isfile(path) and name.startswith('test_') and
+ name.endswith('.py')):
+ submodule_name = os.path.splitext(name)[0]
+ module_name = "{0}.{1}".format(package, submodule_name)
+ __import__(module_name, level=0)
+ module_tests = unittest.findTestCases(sys.modules[module_name])
+ suite.addTest(module_tests)
+ elif os.path.isdir(path):
+ package_name = "{0}.{1}".format(package, name)
+ __import__(package_name, level=0)
+ package_tests = getattr(sys.modules[package_name], 'test_suite')()
+ suite.addTest(package_tests)
+ else:
+ continue
+ return suite
diff --git a/Lib/test/test_import.py b/Lib/test/test_import.py
--- a/Lib/test/test_import.py
+++ b/Lib/test/test_import.py
@@ -1,6 +1,6 @@
import builtins
import imp
-from importlib.test.import_ import test_relative_imports
+from importlib.test.import_ import test_suite as importlib_import_test_suite
from importlib.test.import_ import util as importlib_util
import importlib
import marshal
@@ -694,21 +694,16 @@
self.assertEqual(m.x, 5)
-class RelativeImportFromImportlibTests(test_relative_imports.RelativeImports):
-
- def setUp(self):
- self._importlib_util_flag = importlib_util.using___import__
+def test_main(verbose=None):
+ flag = importlib_util.using___import__
+ try:
importlib_util.using___import__ = True
-
- def tearDown(self):
- importlib_util.using___import__ = self._importlib_util_flag
-
-
-def test_main(verbose=None):
- run_unittest(ImportTests, PycacheTests,
- PycRewritingTests, PathsTests, RelativeImportTests,
- OverridingImportBuiltinTests,
- RelativeImportFromImportlibTests)
+ run_unittest(ImportTests, PycacheTests,
+ PycRewritingTests, PathsTests, RelativeImportTests,
+ OverridingImportBuiltinTests,
+ importlib_import_test_suite())
+ finally:
+ importlib_util.using___import__ = flag
if __name__ == '__main__':
--
Repository URL: http://hg.python.org/cpython
More information about the Python-checkins
mailing list