Function _assert_annotations_equal

mne/tests/test_annotations.py:973–990

(a, b, tol=0, comp_extras_as_str=False)

def _assert_annotations_equal(a, b, tol=0, comp_extras_as_str=False):
    __tracebackhide__ = True
    assert_allclose(a.onset, b.onset, rtol=0, atol=tol, err_msg="onset")
    assert_allclose(a.duration, b.duration, rtol=0, atol=tol, err_msg="duration")
    assert_array_equal(a.description, b.description, err_msg="description")
    assert_array_equal(a.ch_names, b.ch_names, err_msg="ch_names")
    a_orig_time = a.orig_time
    b_orig_time = b.orig_time
    assert a_orig_time == b_orig_time, "orig_time"
    extras_columns = a._extras_columns.union(b._extras_columns)
    for col in extras_columns:
        for i, extra in enumerate(a.extras):
            exa = extra.get(col, None)
            exb = b.extras[i].get(col, None)
            if comp_extras_as_str:
                exa = str(exa) if exa is not None else ""
                exb = str(exb) if exb is not None else ""
            assert exa == exb, f"extras[{i}][{col}]"


_ORIG_TIME = datetime.fromtimestamp(1038942071.7201, timezone.utc)