def my_pipeline(): op1 = ContainerOp(name="op1", image="image") op2 = ContainerOp(name="op2", image="image").after(op1) op3 = ContainerOp(name="op3", image="image") vol1 = PipelineVolume(name="pipeline-volume") vol2 = vol1.after(op1) vol3 = vol2.after(op2) vol4 = vol3.after(op1, op2) vol5 = vol4.after(op3) self.assertEqual(vol1.dependent_names, []) self.assertEqual(vol2.dependent_names, [op1.name]) self.assertEqual(vol3.dependent_names, [op2.name]) self.assertEqual(sorted(vol4.dependent_names), [op1.name, op2.name]) self.assertEqual(sorted(vol5.dependent_names), [op1.name, op2.name, op3.name])
def test_after_method(self): """Test the after method.""" with Pipeline("somename") as p: op1 = ContainerOp(name="op1", image="image") op2 = ContainerOp(name="op2", image="image").after(op1) op3 = ContainerOp(name="op3", image="image") vol1 = PipelineVolume(name="pipeline-volume") vol2 = vol1.after(op1) vol3 = vol2.after(op2) vol4 = vol3.after(op1, op2) vol5 = vol4.after(op3) self.assertEqual(vol1.dependent_names, []) self.assertEqual(vol2.dependent_names, [op1.name]) self.assertEqual(vol3.dependent_names, [op2.name]) self.assertEqual(sorted(vol4.dependent_names), [op1.name, op2.name]) self.assertEqual(sorted(vol5.dependent_names), [op1.name, op2.name, op3.name])