def test_invalid_target(self, target):
pipeline = CompVizPipeline()
pipeline.add_operator('op0', TestOperator())
pipeline.add_operator('op1', TestOperator())
with pytest.raises(ValueError) as e:
pipeline.move_operator('op0', target)
assert_terms_in_exception(e, ['invalid', 'target'])
def test_happy_path(self, idx):
op_to_move_name = 'op1'
op_to_move = TestOperator()
target = 2
pipeline = CompVizPipeline()
pipeline.add_operator('op0', TestOperator())
pipeline.add_operator(op_to_move_name, op_to_move)
pipeline.add_operator('op2', TestOperator())
pipeline.move_operator(idx, target)
assert list(pipeline.operators.values())[target] == op_to_move
assert list(pipeline.operators.keys())[target] == op_to_move_name
