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]
