def METHOD_NAME(shape, layout, element_map, dev, dtype=np.uint8):
    pipe = element_extract_pipe(shape, layout, element_map, dev, dtype)
    pipe.build()
    for i in range(10):
        results = pipe.run()
        input = results[0]
        elements = results[1:]
        for i in range(batch_size):
            for j, idx in enumerate(element_map):
                assert elements[j][i].layout() == layout[1:]
                expected = to_array(input[i])[idx]
                obtained = to_array(elements[j][i])
                np.testing.assert_array_equal(expected, obtained)