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))