diff options
Diffstat (limited to 'Lib/importlib/test/__init__.py')
| -rw-r--r-- | Lib/importlib/test/__init__.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/Lib/importlib/test/__init__.py b/Lib/importlib/test/__init__.py index f9dd57e..bda33e6 100644 --- a/Lib/importlib/test/__init__.py +++ b/Lib/importlib/test/__init__.py @@ -9,7 +9,8 @@ def test_suite(package=__package__, directory=os.path.dirname(__file__)): if name.startswith('.'): continue path = os.path.join(directory, name) - if os.path.isfile(path) and name.startswith('test_'): + 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) @@ -20,6 +21,8 @@ def test_suite(package=__package__, directory=os.path.dirname(__file__)): __import__(package_name, level=0) package_tests = getattr(sys.modules[package_name], 'test_suite')() suite.addTest(package_tests) + else: + continue return suite |