def test_output_transformer_should_zip_data_input_and_expected_output_in_the_transformed_output( tmpdir: LocalPath): pipeline = Pipeline([MultiplyBy2OutputTransformer()]) pipeline, new_data_container = pipeline.handle_fit_transform( DataContainer(data_inputs=[1, 2, 3], current_ids=[0, 1, 2], expected_outputs=[2, 3, 4]), ExecutionContext(tmpdir)) assert new_data_container.data_inputs == [2, 4, 6] assert new_data_container.expected_outputs == [4, 6, 8]
def test_output_transformer_should_zip_data_input_and_expected_output_in_the_transformed_output( tmpdir): pipeline = Pipeline([MultiplyBy2OutputTransformer()]) pipeline, new_data_container = pipeline.handle_fit_transform( DataContainer(current_ids=[0, 1, 2], data_inputs=[1, 2, 3], expected_outputs=[2, 3, 4]), ExecutionContext.create_from_root(pipeline, ExecutionMode.FIT_TRANSFORM, tmpdir)) assert new_data_container.data_inputs == [2, 4, 6] assert new_data_container.expected_outputs == [4, 6, 8]