def test_rename_keeps_saving_order(self): pipeline = CompVizPipeline() pipeline.add_operator('op1', TestOperator()) pipeline.add_operator('op2', TestOperator()) pipeline.add_operator('op3', TestOperator()) pipeline.rename_operator('op2', 'renamed') stream = StringIO() pipeline.save(stream) stream.seek(0) config = yaml.safe_load(stream) operators_order = [ op_config['name'] for op_config in config['pipeline'] ] assert operators_order == ['op1', 'renamed', 'op3']
def test_pipeline_save_writes(self): pipeline = CompVizPipeline() output_stream = Mock() output_stream.write = Mock() pipeline.save(output_stream) output_stream.write.assert_called()
def test_pipeline_save_runs(self): pipeline = CompVizPipeline() pipeline.save(StringIO())