def test_resume_simple(tmpdir):
try:
ncores = 4
m = LocalManager(['--log', 'debug', '--wd', tmpdir, '--report-format', 'json', '--nodes', str(ncores)],
{'wdir': str(tmpdir)})
its = 10
job_req = {
'name': 'sleep',
'execution': {
'exec': '/bin/sleep',
'args': [ '4s' ],
'stdout': 'out',
},
'iteration': { 'stop': its },
'resources': { 'numCores': { 'exact': 1 } }
}
jobs = Jobs().add_std(job_req)
job_ids = m.submit(jobs)
# because job iterations executes in order, after finish of 4th iteration, the three previous should also finish
m.wait4('sleep:3')
jinfos = m.info_parsed(job_ids, withChilds=True)
assert jinfos
jinfo = jinfos['sleep']
assert all((jinfo.iterations, jinfo.iterations.get('start', -1) == 0,
jinfo.iterations.get('stop', 0) == its, jinfo.iterations.get('total', 0) == its,
jinfo.iterations.get('finished', 0) == ncores, jinfo.iterations.get('failed', -1) == 0)), str(jinfo)
assert len(jinfo.childs) == its
for iteration in range(its):
job_it = jinfo.childs[iteration]
exp_status = ['SUCCEED']
if iteration > 3:
exp_status = ['EXECUTING', 'SCHEDULED', 'QUEUED']
assert all((job_it.iteration == iteration,
job_it.name == '{}:{}'.format('sleep', iteration),
job_it.status in exp_status)),\
f"{job_it.iteration} != {iteration}, {job_it.name} != {'{}:{}'.format('sleep', iteration)}, {job_it.status} != {exp_status}"
# kill process
m.kill_manager_process()
m.cleanup()
ncores = 4
m = LocalManager(['--log', 'debug', '--wd', tmpdir, '--report-format', 'json', '--nodes', str(ncores),
'--resume', tmpdir],
{'wdir': str(tmpdir)})
m.wait4all()
jinfos = m.info_parsed(job_ids, withChilds=True)
assert jinfos
jinfo = jinfos['sleep']
assert all((jinfo.iterations, jinfo.iterations.get('start', -1) == 0,
jinfo.iterations.get('stop', 0) == its, jinfo.iterations.get('total', 0) == its,
jinfo.iterations.get('finished', 0) == its, jinfo.iterations.get('failed', -1) == 0)), str(jinfo)
assert len(jinfo.childs) == its
for iteration in range(its):
job_it = jinfo.childs[iteration]
assert all((job_it.iteration == iteration,
job_it.name == '{}:{}'.format('sleep', iteration),
job_it.status == 'SUCCEED')), \
f"{job_it.iteration} != {iteration}, {job_it.name} != {'{}:{}'.format('sleep', iteration)}, {job_it.status} != SUCCEED"
finally:
if m:
m.finish()
m.cleanup()
df = pd.read_csv("accessible_camp_ipc.csv")
camp_coords_df = df[['lon', 'lat']]
coords = camp_coords_df.to_numpy()
# obtain coordinates of selected camps
popu = coords[X_1D, :]
MOO_log(msg="{}".format("#" * 50))
MOO_log(msg="locations of camp Z:\n\t{}".format(popu))
MOO_log(msg="corresponding objective values:\n\t{}".format(res.pop.get("F")))
out_F = res.pop.get("F")
out_F[:, 1] = -out_F[:, 1]
out_F[:, -1] = -out_F[:, -1]
output = np.hstack([popu, out_F])
with open("population.csv", "w", newline="") as file:
writer = csv.writer(file, delimiter=",")
writer.writerow(["lon", "lat", "obj_1", "obj_2", "obj_3", "obj_4", "obj_5"]) # header
writer.writerows(output)
MOO_log(msg="The output is stored in {}/population.csv\n".format(work_dir))
if USE_PJ is True:
QCG_MANAGER.finish()
QCG_MANAGER.kill_manager_process()
QCG_MANAGER.cleanup()
end_time = time.monotonic()
print('Duration:\t{}'.format(timedelta(seconds=end_time - start_time)))
cmd = '%s run_mscale.py --submodel %s --data_dir=%s --instance_index %d --coupling_type %s --num_instances %d --weather_coupling %s' % (
PYTHON_CMD, SUBMODEL, DATA_DIR, INSTANCE_INDEX, COUPLING_TYPE,
NUM_INSTANCES, WEATHER_COUPLING)
print("\tAdd job with cmd = %s" % (cmd))
TaskID = 'TaskID%d_%s' % (INSTANCE_INDEX + 1, SUBMODEL)
stderr = 'log_task/%s_${jname}__${uniq}.stderr' % (TaskID)
stdout = 'log_task/%s_${jname}__${uniq}.stdout' % (TaskID)
jobs.add(name=TaskID,
exec='bash',
args=['-c', cmd],
stdout=stdout,
stderr=stderr,
numCores={'exact': INSTANCE_CORES},
model='default')
INSTANCE_INDEX = INSTANCE_INDEX + 1
ids = m.submit(jobs)
# wait until submited jobs finish
m.wait4(ids)
# get detailed information about submited and finished jobs
print("jobs details:\n%s\n" % str(m.info(ids)))
m.finish()
m.kill_manager_process()
m.cleanup()