Function test_output_layout

dali/test/python/operator_2/test_python_function.py:568–589

()

def test_output_layout():
    pipe = CommonPipeline(1, 1, 0, 999, images_dir)
    with pipe:
        images, _ = pipe.load()
        out1, out2 = fn.python_function(
            images, function=lambda x: (x, x.mean(2)), num_outputs=2, output_layouts=["ABC", "DE"]
        )
        out3, out4 = fn.python_function(
            images, function=lambda x: (x, x / 2), num_outputs=2, output_layouts="FGH"
        )
        out5, out6 = fn.python_function(
            images, function=lambda x: (x, x / 2), num_outputs=2, output_layouts=["IJK"]
        )

        pipe.set_outputs(out1, out2, out3, out4, out5, out6)
    out1, out2, out3, out4, out5, out6 = pipe.run()
    assert out1.layout() == "ABC"
    assert out2.layout() == "DE"
    assert out3.layout() == "FGH"
    assert out4.layout() == "FGH"
    assert out5.layout() == "IJK"
    assert out6.layout() == ""


@raises(RuntimeError, "*length of*output_layouts*greater than*")