Function compare_traces

dali/test/python/operator_1/test_operator_origin_trace.py:102–110

(dali_tbs, python_tbs, end_glob=origin_trace_glob)

def compare_traces(dali_tbs, python_tbs, end_glob=origin_trace_glob):
    assert len(dali_tbs) == len(python_tbs)
    regex = fnmatch.translate(end_glob(op_mode))
    for dali_tb, python_tb in zip(dali_tbs, python_tbs):
        err = f"Comparing dali_tb:\n{dali_tb}\nvs python_tb:\n{python_tb}"
        # Check if we skipped just one frame in python_tb (2 lines)
        assert dali_tb.count("\n") == python_tb.count("\n") + 2, err
        assert dali_tb.startswith(python_tb), err
        assert re.match(regex, dali_tb), f"{dali_tb} didn't match the expected end traceback"


test_modes = ["dali.fn", "dali.ops"]