Function run_two_outputs

dali/test/python/operator_2/test_python_function.py:348–359

(func)

def run_two_outputs(func):
    pipe = BasicPipeline(BATCH_SIZE, NUM_WORKERS, DEVICE_ID, SEED, images_dir)
    pyfunc_pipe = TwoOutputsPythonOperatorPipeline(
        BATCH_SIZE, NUM_WORKERS, DEVICE_ID, SEED, images_dir, func
    )
    for it in range(ITERS):
        (preprocessed_output,) = pipe.run()
        output1, output2 = pyfunc_pipe.run()
        for i in range(len(output1)):
            pro1, pro2 = func(preprocessed_output.at(i))
            assert numpy.array_equal(output1.at(i), pro1)
            assert numpy.array_equal(output2.at(i), pro2)


def test_split():