def test_happy_path(self, idx): operator_to_delete = 'op_to_delete' operator_to_stay = 'op_to_stay' pipeline = CompVizPipeline() pipeline.add_operator(operator_to_stay, TestOperator()) pipeline.add_operator(operator_to_delete, TestOperator()) pipeline.remove_operator(idx) assert list(pipeline.operators.keys()) == [operator_to_stay]
def test_invalid_idx(self, idx): pipeline = CompVizPipeline() pipeline.add_operator('test_op', TestOperator()) with pytest.raises(ValueError) as e: pipeline.remove_operator(idx) assert_terms_in_exception(e, ['invalid', 'operator'])