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
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)
class TestNode(TestCase): """ """ 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 tearDown(self): """ :return: """ try: shutil.rmtree('a') shutil.rmtree('b') shutil.rmtree('c') shutil.rmtree('d') os.remove('test_dag.dag') except: pass def test__init__(self): pass def test__str__(self): pass def test__repr__(self): pass def test_job(self): """ :return: """ pass # dag = DAG('test_dag') # dag.add_node(self.node_a) # self.node_b.pre_script = 'pre' # self.node_c.post_script = 'post' # self.node_c.post_script_args = 'arg1 arg2' # try: # dag.submit() # except: # pass # expected = 'JOB b b/b.job\n' \ # 'JOB c c/c.job\n' \ # 'JOB d d/d.job\n' \ # 'JOB a a/a.job\n\n' \ # 'SCRIPT PRE b pre\n' \ # 'SCRIPT POST c post arg1 arg2\n\n' \ # 'PARENT b CHILD d\n' \ # 'PARENT c CHILD d\n' \ # 'PARENT a CHILD b c\n\n' # # with open(dag.dag_file, 'r') as dag_file: # actual = dag_file.read() # msg = 'testing that the dag file is created properly when the dag is submitted' # self.assertEqual(expected, actual, '%s\nExpected: \n%s\nActual: \n%s\n|' % (msg, expected, actual)) def test_pre_script(self): pass def test_post_script(self): pass def test_parent_nodes(self): pass def test_child_nodes(self): pass def test_retry(self): pass def test_add_parent(self): pass def test_remove_parent(self): pass def test_add_child(self): pass def test_remove_child(self): pass def test_add(self): pass def test_remove(self): pass def test_get_all_family_nodes(self): pass def test_get_all_ancestors(self): pass def test_get_all_descendants(self): pass def test_link_parent_nodes(self): pass def test_link_child_nodes(self): pass def test_list_relations(self): pass 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 test_list_options(self): pass def test_get_child_names(self): pass def test_get_parent_names(self): pass def test_get_names(self): pass