def test_trackers(self):
# add the workflow
structure = self.struct_si
my_wf = get_wf(structure,
"optimize_only.yaml",
vis=MPRelaxSet(structure, force_gamma=True),
common_params={"vasp_cmd": VASP_CMD})
if not VASP_CMD:
my_wf = use_fake_vasp(my_wf, ref_dirs_si)
else:
my_wf = use_custodian(my_wf)
my_wf = add_trackers(my_wf)
self.lp.add_wf(my_wf)
# run the workflow
rapidfire(self.lp, fworker=_fworker)
for x in self.lp.get_tracker_data(1):
for t in x["trackers"]:
self.assertGreater(len(t.content.split("\n")), 20)
wf = self.lp.get_wf_by_fw_id(1)
self.assertTrue(all([s == 'COMPLETED' for s in wf.fw_states.values()]))
def test_trackers(self):
# add the workflow
structure = self.struct_si
my_wf = get_wf(structure, "optimize_only.yaml",
vis=MPRelaxSet(structure, force_gamma=True),
common_params={"vasp_cmd": VASP_CMD})
if not VASP_CMD:
my_wf = use_fake_vasp(my_wf, ref_dirs_si)
else:
my_wf = use_custodian(my_wf)
my_wf = add_trackers(my_wf)
self.lp.add_wf(my_wf)
# run the workflow
rapidfire(self.lp)
for x in self.lp.get_tracker_data(1):
for t in x["trackers"]:
self.assertGreater(len(t.content.split("\n")), 20)
def test_add_trackers(self):
my_wf = add_trackers(self._copy_wf(self.bs_wf))
for fw in my_wf.fws:
self.assertEqual(len(fw.spec["_trackers"]), 2)
def test_add_trackers(self):
my_wf = add_trackers(self._copy_wf(self.bs_wf))
for fw in my_wf.fws:
self.assertEqual(len(fw.spec["_trackers"]), 2)