
##############################################################################
#
# Copyright (c) 2003-2016 by The University of Queensland
# http://www.uq.edu.au
#
# Primary Business: Queensland, Australia
# Licensed under the Apache License, version 2.0
# http://www.apache.org/licenses/LICENSE-2.0
#
# Development until 2012 by Earth Systems Science Computational Center (ESSCC)
# Development 2012-2013 by School of Earth Sciences
# Development from 2014 by Centre for Geoscience Computing (GeoComp)
#
##############################################################################

from grouptest import *
import os
Import('*')
local_env = env.Clone()

example_files = example_files_allow_mpi + example_files_no_mpi

src_dir = local_env.Dir('.').srcnode().abspath
test_dir= Dir('test', local_env.Dir('.'))
cc=Command(test_dir, [], Mkdir("$TARGET"))
# Execute(Delete(test_dir.srcnode().abspath))
# Execute(Mkdir(test_dir.srcnode().abspath))
#
#   create copy of all non-py files:
#
data_files=[]
for i in example_files:
    if not i.endswith('.py'):
        f_in=File(os.path.join(src_dir,i))
        f_out=File(i,test_dir)
        data_files.append(Command(f_out, f_in, Copy("$TARGET", "$SOURCE")))

# We need this because of cblib.py which needs to be present for other tests
# to work
for i in example_deps:
    f_in=File(os.path.join(src_dir,i))
    f_out=File(i,test_dir)
    data_files.append(Command(f_out, f_in, Copy("$TARGET", "$SOURCE")))
Depends(cc, data_files)

np=int(local_env['ENV']['ESCRIPT_NUM_NODES'])
programs=[]
copies=[]
for i in example_files:
#    if i.endswith('.py') and (np==1 or not i in example_files_no_mpi):
    if i.endswith('.py') and (local_env['mpi']=='none' or not i in example_files_no_mpi):
        f_in=File(os.path.join(src_dir,i))
        f_out=File(i, test_dir)
        c=Command(f_out, f_in, Copy("$TARGET", "$SOURCE"))
        copies+=c
        programs+=local_env.RunPyExample(c)

#env.Alias('build_py_tests', programs)
env.Alias('build_py_tests', data_files)
env.Alias('build_py_tests', copies)
env.Alias('py_tests', programs)
Depends(programs, data_files)
Depends(cc, data_files)

# Add a group of tests
# find all directories:
for single_processor_only in [True, False]:
   dirs=[]
   for i in example_files:
      if i.endswith('.py'): 
         if single_processor_only == (i in example_files_no_mpi):
               d=os.path.split(i)[0]
               if not d in dirs: dirs.append(d)
   for d in dirs:
      runs=[]
      for i in example_files:
          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])
      TestGroups.append(GroupTest("$PYTHONRUNNER ",(),"",os.path.join("$BATCH_ROOT/doc/examples",d),runs, single_processor_only=single_processor_only))

