14adfde8bc82dd39f59e0445588c3e599ada477dJosh Gaoimport os
24adfde8bc82dd39f59e0445588c3e599ada477dJosh Gaoimport sys
34adfde8bc82dd39f59e0445588c3e599ada477dJosh Gaoimport unittest
44adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao
54adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao
64adfde8bc82dd39f59e0445588c3e599ada477dJosh Gaohere = os.path.dirname(__file__)
74adfde8bc82dd39f59e0445588c3e599ada477dJosh Gaoloader = unittest.defaultTestLoader
84adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao
94adfde8bc82dd39f59e0445588c3e599ada477dJosh Gaodef suite():
104adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao    suite = unittest.TestSuite()
114adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao    for fn in os.listdir(here):
124adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao        if fn.startswith("test") and fn.endswith(".py"):
134adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao            modname = "unittest.test." + fn[:-3]
144adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao            __import__(modname)
154adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao            module = sys.modules[modname]
164adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao            suite.addTest(loader.loadTestsFromModule(module))
174adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao    return suite
184adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao
194adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao
204adfde8bc82dd39f59e0445588c3e599ada477dJosh Gaoif __name__ == "__main__":
214adfde8bc82dd39f59e0445588c3e599ada477dJosh Gao    unittest.main(defaultTest="suite")
22