source: inundation/pyvolution/test_all.py @ 2252

Last change on this file since 2252 was 2252, checked in by ole, 18 years ago

Cosmetic

File size: 2.6 KB
Line 
1"""Regression testing framework
2This module will search for scripts in the same directory named
3test_*.py.  Each such script should be a test suite that tests a
4module through PyUnit. This script will aggregate all
5found 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
11import unittest
12import 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_pmesh2domain.py']
20
21
22def 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
44def 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
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    print 'Testing:'
63    for file in files:
64        print '  ' + file
65
66    if globals().has_key('exclude'):
67        for file in exclude:
68            files.remove(file)
69            print 'WARNING: File '+ file + ' excluded from testing'
70
71
72    filenameToModuleName = lambda f: os.path.splitext(f)[0]
73    moduleNames = map(filenameToModuleName, files)
74    modules = map(__import__, moduleNames)
75    load = unittest.defaultTestLoader.loadTestsFromModule
76    return unittest.TestSuite(map(load, modules))
77
78if __name__ == '__main__':
79
80    from os import sep
81
82    #Attempt to compile all extensions
83    execfile('..' + sep + 'utilities' + sep + 'compile.py')
84
85    #FIXME: Temporary measure
86    os.chdir('..' + sep + 'utilities')
87    execfile('compile.py')
88    os.chdir('..' + sep + 'pyvolution')   
89   
90    #FIXME: Temporary measure
91    os.chdir('..' + sep + 'triangle')
92    execfile('compile.py')
93    os.chdir('..' + sep + 'pyvolution')   
94   
95    #os.system('python compile.py')
96
97    #print regressionTest()
98    unittest.main(defaultTest='regressionTest')
Note: See TracBrowser for help on using the repository browser.