__init__.py 584 Bytes
Newer Older
1 2 3 4 5 6 7 8
import os
import sys
import unittest


here = os.path.dirname(__file__)
loader = unittest.defaultTestLoader

9
def suite():
10 11 12 13 14 15 16
    suite = unittest.TestSuite()
    for fn in os.listdir(here):
        if fn.startswith("test") and fn.endswith(".py"):
            modname = "unittest.test." + fn[:-3]
            __import__(modname)
            module = sys.modules[modname]
            suite.addTest(loader.loadTestsFromModule(module))
17
    suite.addTest(loader.loadTestsFromName('unittest.test.testmock'))
18 19 20 21
    return suite


if __name__ == "__main__":
22
    unittest.main(defaultTest="suite")