[1018] | 1 | """Regression testing framework |
---|
| 2 | This module will search for scripts in the same directory named |
---|
| 3 | test_*.py. Each such script should be a test suite that tests a |
---|
| 4 | module through PyUnit. This script will aggregate all |
---|
| 5 | found test suites into one big test suite and run them all at once. |
---|
| 6 | """ |
---|
| 7 | |
---|
| 8 | # Author: Mark Pilgrim |
---|
| 9 | # Modified by Ole Nielsen |
---|
| 10 | |
---|
| 11 | import unittest |
---|
| 12 | import os |
---|
| 13 | |
---|
| 14 | |
---|
| 15 | #List files that should be excluded from the testing process. |
---|
| 16 | #E.g. if they are known to fail and under development |
---|
| 17 | #exclude = ['test_least_squares.py', 'test_cg_solve.py', |
---|
| 18 | # 'test_interpolate_sww.py'] |
---|
| 19 | #exclude = ['test_cg_solve.py'] |
---|
| 20 | |
---|
| 21 | |
---|
| 22 | def get_test_files(path): |
---|
| 23 | |
---|
| 24 | import sys |
---|
| 25 | |
---|
| 26 | files = os.listdir(path) |
---|
| 27 | |
---|
| 28 | #Check sub directories |
---|
| 29 | test_files = [] |
---|
| 30 | for file in files: |
---|
| 31 | if os.path.isdir(file): |
---|
| 32 | sys.path.append(file) |
---|
| 33 | #print 'Recursing into', file |
---|
| 34 | test_files += get_test_files(path + os.sep + file) |
---|
| 35 | elif file[:5] == 'test_' and file[-2:] == 'py': |
---|
| 36 | #print 'Appending', file |
---|
| 37 | test_files.append(file) |
---|
| 38 | else: |
---|
| 39 | pass |
---|
| 40 | return test_files |
---|
| 41 | |
---|
| 42 | |
---|
| 43 | |
---|
| 44 | def regressionTest(): |
---|
| 45 | import sys, os, re, unittest |
---|
| 46 | path = os.path.split(sys.argv[0])[0] or os.getcwd() |
---|
| 47 | |
---|
| 48 | |
---|
| 49 | files = get_test_files(path) |
---|
| 50 | |
---|
| 51 | #print 'Testing:', files |
---|
| 52 | |
---|
| 53 | #test = re.compile('^test_[\w]*.py$', re.IGNORECASE) |
---|
| 54 | #files = filter(test.search, files) |
---|
| 55 | |
---|
| 56 | |
---|
| 57 | try: |
---|
| 58 | files.remove(__file__) #Remove self from list (Ver 2.3. or later) |
---|
| 59 | except: |
---|
| 60 | files.remove('test_all.py') |
---|
| 61 | |
---|
| 62 | |
---|
| 63 | if globals().has_key('exclude'): |
---|
| 64 | for file in exclude: |
---|
| 65 | files.remove(file) |
---|
| 66 | print 'WARNING: File '+ file + ' excluded from testing' |
---|
| 67 | |
---|
| 68 | |
---|
| 69 | filenameToModuleName = lambda f: os.path.splitext(f)[0] |
---|
| 70 | moduleNames = map(filenameToModuleName, files) |
---|
| 71 | modules = map(__import__, moduleNames) |
---|
| 72 | load = unittest.defaultTestLoader.loadTestsFromModule |
---|
| 73 | return unittest.TestSuite(map(load, modules)) |
---|
| 74 | |
---|
| 75 | if __name__ == '__main__': |
---|
| 76 | |
---|
[1087] | 77 | execfile('compile.py') |
---|
| 78 | #os.system('python compile.py') #Attempt to compile all extensions |
---|
[1018] | 79 | |
---|
| 80 | #print regressionTest() |
---|
| 81 | unittest.main(defaultTest='regressionTest') |
---|