def test_should_inverse_transform(): step_cloner = StepClonerForEachDataInput(SomeStepInverseTransform()) step_cloner, processed_outputs = step_cloner.fit_transform([0]) step_cloner = step_cloner.reverse() processed_outputs = step_cloner.inverse_transform(processed_outputs) assert processed_outputs == ['inverse_transform']
def test_step_cloner_should_inverse_transform(): tape = TapeCallbackFunction() p = StepClonerForEachDataInput( Pipeline([FitCallbackStep(tape), MultiplyByN(2)])) data_inputs = _create_data((2, 2)) expected_outputs = _create_data((2, 2)) p, processed_outputs = p.fit_transform(data_inputs, expected_outputs) p = p.reverse() assert np.array_equal(processed_outputs, data_inputs * 2) inverse_processed_outputs = p.inverse_transform(processed_outputs) assert np.array_equal(np.array(inverse_processed_outputs), np.array(data_inputs))