4
4
import yaml
5
5
from ufs_test_utils import get_testcase , write_logfile , rrmdir , machine_check_off
6
6
7
- def rocoto_create_entries (RTPWD ,MACHINE_ID ,INPUTDATA_ROOT ,INPUTDATA_ROOT_WW3 ,INPUTDATA_ROOT_BMIC ,RUNDIR_ROOT ,NEW_BASELINE ,ROCOTO_XML ):
7
+ def rocoto_create_entries (RTPWD ,MACHINE_ID ,INPUTDATA_ROOT ,INPUTDATA_ROOT_WW3 ,INPUTDATA_ROOT_BMIC ,HSD_INPUT_DATA , RUNDIR_ROOT ,NEW_BASELINE ,ROCOTO_XML ):
8
8
"""Generate header information for Rocoto xml file
9
9
10
10
Args:
@@ -13,6 +13,7 @@ def rocoto_create_entries(RTPWD,MACHINE_ID,INPUTDATA_ROOT,INPUTDATA_ROOT_WW3,INP
13
13
INPUTDATA_ROOT (str): Input data directory
14
14
INPUTDATA_ROOT_WW3 (str): WW3 input data directory
15
15
INPUTDATA_ROOT_BMIC (str): BMIC input data directory
16
+ HSD_INPUT_DATA (str): HSD tests input data directory
16
17
RUNDIR_ROOT (str): Test run directory
17
18
NEW_BASELINE (str): Directory for newly generated baselines
18
19
ROCOTO_XML (str): Rocoto .xml filename to write to
@@ -31,6 +32,7 @@ def rocoto_create_entries(RTPWD,MACHINE_ID,INPUTDATA_ROOT,INPUTDATA_ROOT_WW3,INP
31
32
<!ENTITY INPUTDATA_ROOT "{ INPUTDATA_ROOT } ">
32
33
<!ENTITY INPUTDATA_ROOT_WW3 "{ INPUTDATA_ROOT_WW3 } ">
33
34
<!ENTITY INPUTDATA_ROOT_BMIC "{ INPUTDATA_ROOT_BMIC } ">
35
+ <!ENTITY HSD_INPUT_DATA "{ HSD_INPUT_DATA } ">
34
36
<!ENTITY RUNDIR_ROOT "{ RUNDIR_ROOT } ">
35
37
<!ENTITY NEW_BASELINE "{ NEW_BASELINE } ">
36
38
]>
@@ -166,6 +168,7 @@ def write_runtest_env():
166
168
INPUTDATA_ROOT = str (os .getenv ('INPUTDATA_ROOT' ))
167
169
INPUTDATA_ROOT_WW3 = str (os .getenv ('INPUTDATA_ROOT_WW3' ))
168
170
INPUTDATA_ROOT_BMIC = str (os .getenv ('INPUTDATA_ROOT_BMIC' ))
171
+ HSD_INPUT_DATA = str (os .getenv ('HSD_INPUT_DATA' ))
169
172
PATHRT = str (os .getenv ('PATHRT' ))
170
173
PATHTR , tail = os .path .split (PATHRT )
171
174
NEW_BASELINE = str (os .getenv ('NEW_BASELINE' ))
@@ -194,6 +197,7 @@ def write_runtest_env():
194
197
export INPUTDATA_ROOT={ INPUTDATA_ROOT }
195
198
export INPUTDATA_ROOT_WW3={ INPUTDATA_ROOT_WW3 }
196
199
export INPUTDATA_ROOT_BMIC={ INPUTDATA_ROOT_BMIC }
200
+ export HSD_INPUT_DATA={ HSD_INPUT_DATA }
197
201
export PATHRT={ PATHRT }
198
202
export PATHTR={ PATHTR }
199
203
export NEW_BASELINE={ NEW_BASELINE }
@@ -343,6 +347,7 @@ def xml_loop():
343
347
INPUTDATA_ROOT = str (base ['INPUTDATA_ROOT' ])
344
348
INPUTDATA_ROOT_WW3 = str (base ['INPUTDATA_ROOT_WW3' ])
345
349
INPUTDATA_ROOT_BMIC = str (base ['INPUTDATA_ROOT_BMIC' ])
350
+ HSD_INPUT_DATA = str (base ['HSD_INPUT_DATA' ])
346
351
347
352
path = STMP + '/' + USER
348
353
os .makedirs (path , exist_ok = True )
@@ -387,6 +392,7 @@ def xml_loop():
387
392
os .environ ["INPUTDATA_ROOT" ] = INPUTDATA_ROOT
388
393
os .environ ["INPUTDATA_ROOT_WW3" ] = INPUTDATA_ROOT_WW3
389
394
os .environ ["INPUTDATA_ROOT_BMIC" ]= INPUTDATA_ROOT_BMIC
395
+ os .environ ["HSD_INPUT_DATA" ]= HSD_INPUT_DATA
390
396
os .environ ["PARTITION" ] = PARTITION
391
397
os .environ ["SCHEDULER" ] = SCHEDULER
392
398
os .environ ["RTPWD" ] = RTPWD
@@ -395,7 +401,7 @@ def xml_loop():
395
401
JOB_NR = 0
396
402
ROCOTO = True
397
403
ROCOTO_XML = os .getenv ('ROCOTO_XML' )
398
- rocoto_create_entries (RTPWD ,MACHINE_ID ,INPUTDATA_ROOT ,INPUTDATA_ROOT_WW3 ,INPUTDATA_ROOT_BMIC ,RUNDIR_ROOT ,NEW_BASELINE ,ROCOTO_XML )
404
+ rocoto_create_entries (RTPWD ,MACHINE_ID ,INPUTDATA_ROOT ,INPUTDATA_ROOT_WW3 ,INPUTDATA_ROOT_BMIC ,HSD_INPUT_DATA , RUNDIR_ROOT ,NEW_BASELINE ,ROCOTO_XML )
399
405
UFS_TEST_YAML = str (os .getenv ('UFS_TEST_YAML' ))
400
406
with open (UFS_TEST_YAML , 'r' ) as f :
401
407
rt_yaml = yaml .load (f , Loader = yaml .FullLoader )
0 commit comments