def test_use_scratch_dir(self): my_wf = self._copy_wf(self.bs_wf) my_wf = use_custodian(my_wf) my_wf = use_scratch_dir(my_wf, ">>scratch_dir<<") found = 0 for fw in my_wf.fws: for t in fw.tasks: if 'RunVaspCustodian' in str(t): self.assertEqual(t["scratch_dir"], ">>scratch_dir<<") found += 1 self.assertEqual(found, 4)
def test_use_scratch_dir(self): my_wf = self._copy_wf(self.bs_wf) my_wf = use_custodian(my_wf) my_wf = use_scratch_dir(my_wf, ">>scratch_dir<<") found = 0 for fw in my_wf.fws: for t in fw.tasks: if 'RunVaspCustodian' in str(t): self.assertEqual(t["scratch_dir"], ">>scratch_dir<<") found += 1 self.assertEqual(found, 4)