__init__.py 584 B

12345678910111213141516171819202122
  1. import os
  2. import sys
  3. import unittest
  4. here = os.path.dirname(__file__)
  5. loader = unittest.defaultTestLoader
  6. def suite():
  7. suite = unittest.TestSuite()
  8. for fn in os.listdir(here):
  9. if fn.startswith("test") and fn.endswith(".py"):
  10. modname = "unittest.test." + fn[:-3]
  11. __import__(modname)
  12. module = sys.modules[modname]
  13. suite.addTest(loader.loadTestsFromModule(module))
  14. suite.addTest(loader.loadTestsFromName('unittest.test.testmock'))
  15. return suite
  16. if __name__ == "__main__":
  17. unittest.main(defaultTest="suite")