Function check_layout

dali/test/python/test_external_source_parallel_utils.py:221–229

(pipe, layout)

def check_layout(pipe, layout):
    pipe.build()
    capture_processes(pipe._py_pool)
    while True:
        try:
            (res,) = pipe.run()
            assert res.layout() == layout
        except StopIteration:
            break