def run_all_python_modules(files, files_to_be_ignored=()):
suite = unittest.TestSuite()
for python_file in files:
if python_file not in files_to_be_ignored:
python_module = __import__(python_file[:-3])
if python_module.__doc__:
suite.addTest(doctest.DocTestSuite(python_module,
optionflags=FLAGS))
return suite
PYHISTORIAN_DIRNAME = os.path.dirname(os.path.abspath(pyhistorian.__file__))
def run_all_python_modules(files, files_to_be_ignored=()):
suite = unittest.TestSuite()
for python_file in files:
if python_file not in files_to_be_ignored:
python_module = __import__(python_file[:-3])
if python_module.__doc__:
suite.addTest(doctest.DocTestSuite(python_module,
optionflags=FLAGS))
return suite
def run_python_modules_outhere():
all_pyhistorian_python_files = ['pyhistorian.'+filename for filename in os.listdir(PYHISTORIAN_DIRNAME)
if filename.endswith('.py')]
return run_all_python_modules(all_pyhistorian_python_files, ['pyhistorian.__init__.py', 'pyhistorian.plone.py'])
def run_all_python_modules_here():
all_python_files_here = [filename for filename in os.listdir(THIS_DIRNAME)
if filename.endswith('.py')]
return run_all_python_modules(all_python_files_here, ['__init__.py',
'run_specs.py'])
if __name__ == '__main__':
suite = unittest.TestSuite()
suite.addTest(run_all_python_modules_here())
suite.addTest(run_python_modules_outhere())
runner = unittest.TextTestRunner()
result = runner.run(suite)
exit(result.wasSuccessful() and 0 or 1)
