"""test utilities

 (part of web.py)

 """

 import unittest

 import sys, os

 import web

 TestCase = unittest.TestCase

 TestSuite = unittest.TestSuite

 def load_modules(names):

     return [__import__(name, None, None, "x") for name in names]

 def module_suite(module, classnames=None):

     """Makes a suite from a module."""

     if classnames:

         return unittest.TestLoader().loadTestsFromNames(classnames, module)

     elif hasattr(module, 'suite'):

         return module.suite()

     else:

         return unittest.TestLoader().loadTestsFromModule(module)

 def doctest_suite(module_names):

     """Makes a test suite from doctests."""

     import doctest

     suite = TestSuite()

     for mod in load_modules(module_names):

         suite.addTest(doctest.DocTestSuite(mod))

     return suite

 def suite(module_names):

     """Creates a suite from multiple modules."""

     suite = TestSuite()

     for mod in load_modules(module_names):

         suite.addTest(module_suite(mod))

     return suite

 def runTests(suite):

     runner = unittest.TextTestRunner()

     return runner.run(suite)

 def main(suite=None):

     if not suite:

         main_module = __import__('__main__')

         # allow command line switches

         args = [a for a in sys.argv[1:] if not a.startswith('-')]

         suite = module_suite(main_module, args or None)

     result = runTests(suite)

     sys.exit(not result.wasSuccessful())