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