From fed09c7435e7b19c59aad9b88ddb0c76912c53e9 Mon Sep 17 00:00:00 2001 From: Waylan Limberg Date: Tue, 14 Oct 2008 23:05:59 -0400 Subject: Cleaned up regression test suite code. --- regression-tests.py | 30 ++++++++++++++++-------------- 1 file changed, 16 insertions(+), 14 deletions(-) (limited to 'regression-tests.py') diff --git a/regression-tests.py b/regression-tests.py index 41aa34c..c8f784c 100755 --- a/regression-tests.py +++ b/regression-tests.py @@ -137,19 +137,21 @@ class TestTreap(unittest.TestCase): [('first', 'This'), ('second', 'is'), ('third', 'a'), ('fourth', 'self'), ('seventh','.'), ('fifth', 'test')]) -suite = unittest.TestSuite() -suite.addTest(unittest.makeSuite(TestMarkdownParser)) -suite.addTest(unittest.makeSuite(TestHtmlStash)) -suite.addTest(unittest.makeSuite(TestTreap)) - -for filename in os.listdir('markdown_extensions'): - if filename.endswith('.py'): - module = 'markdown_extensions.%s' % filename[:-3] - try: - suite.addTest(DocTestSuite(module)) - except: ValueError - # No tests +def suite(): + """ Build a test suite of the above tests and extension doctests. """ + suite = unittest.TestSuite() + suite.addTest(unittest.makeSuite(TestMarkdownParser)) + suite.addTest(unittest.makeSuite(TestHtmlStash)) + suite.addTest(unittest.makeSuite(TestTreap)) + + for filename in os.listdir('markdown_extensions'): + if filename.endswith('.py'): + module = 'markdown_extensions.%s' % filename[:-3] + try: + suite.addTest(DocTestSuite(module)) + except: ValueError + # No tests + return suite if __name__ == '__main__': - #unittest.main() - unittest.TextTestRunner(verbosity=2).run(suite) + unittest.TextTestRunner(verbosity=2).run(suite()) -- cgit v1.2.3