Function check_output

dali/test/python/test_utils.py:529–555

Checks the outputs of the pipeline. `outputs` return value from pipeline `run` `ref_out` a batch or tuple of batches `ref_is_list_of_outputs` only meaningful when there's just one output - if True, ref_out is a one-element list containing a single batch f

(outputs, ref_out, ref_is_list_of_outputs=None)

def check_output(outputs, ref_out, ref_is_list_of_outputs=None):
    """Checks the outputs of the pipeline.

    `outputs`
        return value from pipeline `run`
    `ref_out`
        a batch or tuple of batches
    `ref_is_list_of_outputs`
        only meaningful when there's just one output - if True, ref_out is a one-element
        list containing a single batch for output 0; otherwise ref_out _is_ a batch
    """
    import_numpy()
    if ref_is_list_of_outputs is None:
        ref_is_list_of_outputs = len(outputs) > 1

    assert ref_is_list_of_outputs or (len(outputs) == 1)

    for idx in range(len(outputs)):
        out = outputs[idx]
        ref = ref_out[idx] if ref_is_list_of_outputs else ref_out
        out = out.as_cpu()
        for i in range(len(out)):
            if not np.array_equal(out[i], ref[i]):
                print("Mismatch at sample", i)
                print("Out: ", out.at(i))
                print("Ref: ", ref[i])
            assert np.array_equal(out[i], ref[i])


def dali_type(t):