1 |
|
2 |
######################################################## |
3 |
# |
4 |
# Copyright (c) 2003-2009 by University of Queensland |
5 |
# Earth Systems Science Computational Center (ESSCC) |
6 |
# http://www.uq.edu.au/esscc |
7 |
# |
8 |
# Primary Business: Queensland, Australia |
9 |
# Licensed under the Open Software License version 3.0 |
10 |
# http://www.opensource.org/licenses/osl-3.0.php |
11 |
# |
12 |
######################################################## |
13 |
|
14 |
from grouptest import * |
15 |
import os |
16 |
Import('*') |
17 |
example_files = example_files_allow_mpi + example_files_no_mpi |
18 |
|
19 |
|
20 |
local_env = clone_env(env) |
21 |
src_dir = local_env.Dir('.').srcnode().abspath |
22 |
test_dir= Dir('test', local_env.Dir('.')) |
23 |
cc=Command(test_dir, [], Mkdir("$TARGET")) |
24 |
# Execute(Delete(test_dir.srcnode().abspath)) |
25 |
# Execute(Mkdir(test_dir.srcnode().abspath)) |
26 |
# |
27 |
# create copy of all non-py files: |
28 |
# |
29 |
data_files=[] |
30 |
for i in example_files: |
31 |
if not i.endswith('.py'): |
32 |
f_in=File(os.path.join(src_dir,i)) |
33 |
f_out=File(i,test_dir) |
34 |
data_files.append(Command(f_out, f_in, Copy("$TARGET", "$SOURCE"))) |
35 |
|
36 |
#We need this because of cblib.py which needs to be present for other tests to work |
37 |
for i in example_deps: |
38 |
f_in=File(os.path.join(src_dir,i)) |
39 |
f_out=File(i,test_dir) |
40 |
data_files.append(Command(f_out, f_in, Copy("$TARGET", "$SOURCE"))) |
41 |
Depends(cc, data_files) |
42 |
|
43 |
np=int(local_env['ENV']['ESCRIPT_NUM_NODES']) |
44 |
programs=[] |
45 |
copies=[] |
46 |
for i in example_files: |
47 |
if i.endswith('.py') and (np==1 or not i in example_files_no_mpi): |
48 |
f_in=File(os.path.join(src_dir,i)) |
49 |
f_out=File(i, test_dir) |
50 |
c=Command(f_out, f_in, Copy("$TARGET", "$SOURCE")) |
51 |
copies+=c |
52 |
programs+=local_env.RunPyUnitTest(c) |
53 |
#env.Alias('build_py_tests', programs) |
54 |
env.Alias('build_py_tests', data_files) |
55 |
env.Alias('build_py_tests', copies) |
56 |
env.Alias('py_tests', programs) |
57 |
Depends(programs, env['pyinstall']) |
58 |
Depends(programs, data_files) |
59 |
Depends(cc, data_files) |
60 |
# Add a group of tests |
61 |
# find all directories: |
62 |
for single_processor_only in [True, False]: |
63 |
dirs=[] |
64 |
for i in example_files: |
65 |
if i.endswith('.py'): |
66 |
if single_processor_only == (i in example_files_no_mpi): |
67 |
d=os.path.split(i)[0] |
68 |
if not d in dirs: dirs.append(d) |
69 |
for d in dirs: |
70 |
runs=[] |
71 |
for i in example_files: |
72 |
if i.startswith(d) and i.endswith('.py') and (single_processor_only == (i in example_files_no_mpi)): runs.append(os.path.split(i)[1]) |
73 |
TestGroups.append(GroupTest("$PYTHONRUNNER ",(),"",os.path.join("$BUILD_DIR/doc/examples/test",d),runs, single_processor_only=single_processor_only)) |