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")