コード例 #1
0
ファイル: test_dags.py プロジェクト: opentechfn/polyaxon
 def test_get_orphan_nodes(self):
     assert dags.get_orphan_ops(self.dag0_processed) == {3, 19}
     assert dags.get_orphan_ops(self.dag1_processed) == {3}
     assert dags.get_orphan_ops(self.dag2_processed) == {4, 8, 10, 11, 28}
     assert dags.get_orphan_ops(self.dag3_processed) == set([])
     assert dags.get_orphan_ops(self.dag4_processed) == set([])
     assert dags.get_orphan_ops(self.dag5_processed) == set([])
     assert dags.get_orphan_ops(self.cycle1_processed) == set([])
     assert dags.get_orphan_ops(self.cycle2_processed) == {8, 10, 11}
コード例 #2
0
ファイル: dag.py プロジェクト: ishandutta2007/polyaxon
 def get_orphan_ops(self, dag=None):
     """Get orphan ops for given dag."""
     return dags.get_orphan_ops(dag or self.dag)