Do not execute the .pyc/.pyo files as well as the .py files.

This commit is contained in:
Brett Cannon 2009-02-01 03:51:54 +00:00
parent 4ee2cdaf65
commit ae9ad186d0
1 changed files with 4 additions and 1 deletions

View File

@ -9,7 +9,8 @@ def test_suite(package=__package__, directory=os.path.dirname(__file__)):
if name.startswith('.'): if name.startswith('.'):
continue continue
path = os.path.join(directory, name) 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] submodule_name = os.path.splitext(name)[0]
module_name = "{0}.{1}".format(package, submodule_name) module_name = "{0}.{1}".format(package, submodule_name)
__import__(module_name, level=0) __import__(module_name, level=0)
@ -20,6 +21,8 @@ def test_suite(package=__package__, directory=os.path.dirname(__file__)):
__import__(package_name, level=0) __import__(package_name, level=0)
package_tests = getattr(sys.modules[package_name], 'test_suite')() package_tests = getattr(sys.modules[package_name], 'test_suite')()
suite.addTest(package_tests) suite.addTest(package_tests)
else:
continue
return suite return suite