Example #1
0
 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}
Example #2
0
 def get_orphan_ops(self, dag=None):
     """Get orphan ops for given dag."""
     return dags.get_orphan_ops(dag or self.dag)