def test_substitution_list(self):
subst_list = SubstitutionList()
subst_list.addItem("Key", "Value", "Doc String")
self.assertEqual(len(subst_list), 1)
with self.assertRaises(IndexError):
item = subst_list["1"]
with self.assertRaises(IndexError):
item = subst_list[2]
key, value, doc_string = subst_list[0]
self.assertEqual(key, "Key")
self.assertEqual(value, "Value")
self.assertEqual(doc_string, "Doc String")
def test_substitution_list(self):
subst_list = SubstitutionList()
subst_list.addItem("Key", "Value", "Doc String")
self.assertEqual(len(subst_list), 1)
with self.assertRaises(IndexError):
item = subst_list["1"]
with self.assertRaises(IndexError):
item = subst_list[2]
key, value, doc_string = subst_list[0]
self.assertEqual(key, "Key")
self.assertEqual(value, "Value")
self.assertEqual(doc_string, "Doc String")
def run_all(self):
forward_model = self.set_up_forward_model(range(len(joblist)))
umask = 0
global_args = SubstitutionList()
forward_model.formatted_fprintf(
os.getcwd(),
global_args,
umask)
self.verify_json_dump(range(len(joblist)), global_args, umask)
def test_no_jobs(self):
with TestAreaContext("python/job_queue/forward_model_no_jobs"):
forward_model = self.set_up_forward_model([])
umask = 4
global_args = SubstitutionList()
forward_model.formatted_fprintf(
os.getcwd(),
global_args,
umask)
self.verify_json_dump([], global_args, umask)
def test_one_job(self):
with TestAreaContext("python/job_queue/forward_model_one_job"):
for i in range(len(joblist)):
forward_model = self.set_up_forward_model([i])
umask = 11
global_args = SubstitutionList()
forward_model.formatted_fprintf(
os.getcwd(),
global_args,
umask)
self.verify_json_dump([i], global_args, umask)
def test_workflow_run(self):
with TestAreaContext("python/job_queue/workflow") as work_area:
WorkflowCommon.createExternalDumpJob()
joblist = WorkflowJoblist()
self.assertTrue(joblist.addJobFromFile("DUMP", "dump_job"))
self.assertTrue("DUMP" in joblist)
workflow = Workflow("dump_workflow", joblist)
self.assertTrue(len(workflow), 2)
context = SubstitutionList()
context.addItem("<PARAM>", "text")
self.assertTrue(workflow.run(None, verbose=True, context=context))
with open("dump1", "r") as f:
self.assertEqual(f.read(), "dump_text_1")
with open("dump2", "r") as f:
self.assertEqual(f.read(), "dump_text_2")
