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)
__import__(package_name, level=0)
package_tests = getattr(sys.modules[package_name], 'test_suite')()
suite.addTest(package_tests)
+ else:
+ continue
return suite