Function dummy_annotation_file

mne/tests/test_annotations.py:997–1046

Create csv file for testing.

(tmp_path_factory, ch_names, fmt, with_extras)

@pytest.fixture(scope="function", params=("ch_names", "fmt", "with_extras"))
def dummy_annotation_file(tmp_path_factory, ch_names, fmt, with_extras):
    """Create csv file for testing."""
    extras_row0 = {"foo1": 1, "foo2": 1.1, "foo3": "a", "foo4": None}
    if fmt == "csv":
        content = (
            "onset,duration,description\n"
            "2002-12-03 19:01:11.720100,1.0,AA\n"
            "2002-12-03 19:01:20.720100,2.425,BB"
        )
    elif fmt == "txt":
        content = (
            "# MNE-Annotations\n"
            "# orig_time : 2002-12-03 19:01:11.720100\n"
            "# onset, duration, description\n"
            "0, 1, AA \n"
            "9, 2.425, BB"
        )
    else:
        assert fmt == "fif"
        extras = [extras_row0, None] if with_extras else None
        content = Annotations(
            [0, 9], [1, 2.425], ["AA", "BB"], orig_time=_ORIG_TIME, extras=extras
        )

    if ch_names:
        if isinstance(content, Annotations):
            # this is a bit of a hack but it works
            content.ch_names[:] = ((), ("MEG0111", "MEG2563"))
        else:
            content = content.splitlines()
            content[-3] += ",ch_names"
            content[-2] += ","
            content[-1] += ",MEG0111:MEG2563"
            content = "\n".join(content)
    if with_extras and fmt != "fif":
        content = content.splitlines()
        content[-3] += "," + ",".join(extras_row0.keys())
        content[-2] += "," + ",".join(
            ["" if v is None else str(v) for v in extras_row0.values()]
        )
        content[-1] += ",,,,"
        content = "\n".join(content)

    fname = tmp_path_factory.mktemp("data") / f"annotations-annot.{fmt}"
    if isinstance(content, str):
        with open(fname, "w") as f:
            f.write(content)
    else:
        content.save(fname)
    return fname


@pytest.mark.filterwarnings("ignore:.*heterogeneous dtypes.*")