Showing the changes to python (1 of 1 file). Show all
Unified · Split
| @@ -1,24 +1,5 @@ | |||
| 1 | #! /usr/bin/env python | 1 | #! /usr/bin/env python |
| 2 | 2 | ||
| 3 | if __name__ == '__main__': | 3 | if __name__ == '__main__': |
| 4 | import os, sys, unittest | 4 | import nose2.main |
| 5 | def load_test(): | 5 | nose2.discover() |
| 6 | # get relevant files | ||
| 7 | curdir = os.path.dirname(sys.argv[0]) | ||
| 8 | if curdir == '': curdir = '.' | ||
| 9 | files = os.listdir(curdir) | ||
| 10 | modfiles = filter (lambda y: y.endswith('.py'), files) | ||
| 11 | modfiles = filter (lambda f: f.startswith('test_'), modfiles) | ||
| 12 | modfiles = filter (lambda y: not 'beattracking' in y, modfiles) | ||
| 13 | modfiles = filter (lambda y: not 'hist' in y, modfiles) | ||
| 14 | modfiles = filter (lambda y: not 'scale' in y, modfiles) | ||
| 15 | modfiles = filter (lambda y: not 'peakpicker' in y, modfiles) | ||
| 16 | # get module names | ||
| 17 | modnames = map (lambda x: os.path.splitext(x)[0], modfiles) | ||
| 18 | # import them | ||
| 19 | modules = map (__import__, modnames) | ||
| 20 | # create a test suites from the imported module | ||
| 21 | load_from_module = unittest.defaultTestLoader.loadTestsFromModule | ||
| 22 | tests = map(load_from_module, modules) | ||
| 23 | return unittest.TestSuite(tests) | ||
| 24 | unittest.main(defaultTest = 'load_test') | ||