def test_pipe_mapper():
df = pd.DataFrame({'t1': [1, 2, 3], 't2': [-1, -2, -3]})
result_df = pd.DataFrame({
't5': [5, 5, 5],
't4': [4, 16, 36]
}).sort_index(axis=1)
output_df = test.process(df)['output'].sort_index(axis=1)
assert output_df.equals(result_df)
def test_pipe_mapper():
df = pd.DataFrame(
{
't1': [1, 2, 3],
't2': [-1, -2, -3]
}
)
result_df = pd.DataFrame(
{
't5': [5, 5, 5],
't4': [4, 16, 36]
}
).sort_index(axis=1)
output_df = test.process(df)['output'].sort_index(axis=1)
assert output_df.equals(result_df)
def test_pipe_mapper():
df = pd.DataFrame(
{
't1': [1, 2, 3],
't2': [-1, -2, -3]
}
)
result_df = pd.DataFrame(
{
't5': [5, 5, 5],
't4': [4, 16, 36],
}
).sort_index(axis=1)
test.append(ConstantDataSource, construct_arguments=[df])
nodes = test.build_process_graph().nodes()
output_df = test.process(output_channels=['temp']).sort_index(axis=1)
assert output_df.equals(result_df)
