def setUp(self): """ :return: """ self.job_a = Job('a', Templates.base) self.job_b = Job('b', Templates.base) self.job_c = Job('c', Templates.base) self.job_d = Job('d', Templates.base) self.node_a = Node(self.job_a) self.node_b = Node(self.job_b) self.node_c = Node(self.job_c) self.node_d = Node(self.job_d) self.node_a.add_child(self.node_b) self.node_a.add_child(self.node_c) self.node_d.add_parent(self.node_b) self.node_d.add_parent(self.node_c)
def test_list_scripts(self): node = Node(self.job_a, post_script='script', post_script_args=' '.join(('arg1', 'arg2', 'arg3'))) expected = 'SCRIPT POST a script arg1 arg2 arg3\n' actual = node.list_scripts() msg = 'testing that post script string is formatted correctly' self.assertEqual( expected, actual, '%s\nExpected: %s\nActual: %s\n' % (msg, expected, actual))
def condorpy_node(self): if not hasattr(self, '_condorpy_node'): condorpy_node = Node(job=self.job, pre_script=self.pre_script, pre_script_args=self.pre_script_args, post_script=self.post_script, post_script_args=self.post_script_args, variables=self.variables, priority=self.priority, category=self.category, retry=self.retry, pre_skip=self.pre_skip, abort_dag_on=self.abort_dag_on, abort_dag_on_return_value=self.abort_dag_on_return_value, dir=self.dir, noop=self.noop, done=self.done ) self._condorpy_node = condorpy_node return self._condorpy_node