Skip to content

Commit

Permalink
Add test
Browse files Browse the repository at this point in the history
  • Loading branch information
withmywoessner committed Jan 22, 2025
1 parent 5c2e8a7 commit 7100772
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions mne/tests/test_annotations.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,7 @@
first_samps = pytest.mark.parametrize("first_samp", (0, 10000))
edf_reduced = data_path / "EDF" / "test_reduced.edf"
edf_annot_only = data_path / "EDF" / "SC4001EC-Hypnogram.edf"
wfdb_dir = data_path / "wfbd"


needs_pandas = pytest.mark.skipif(not check_version("pandas"), reason="Needs pandas")
Expand Down Expand Up @@ -415,6 +416,14 @@ def test_read_edf_annotations(fname, n_annot):
annot = read_annotations(fname)
assert len(annot) == n_annot

@testing.requires_testing_data
def test_read_wfdb_annotations():
"""Test reading WFDB annotations."""
fname = wfdb_dir / "chb06_04.edf.seizures"
annot = read_annotations(fname)
# Verify that [ and ] are in description
assert "[" in annot.description
assert "]" in annot.description

@first_samps
def test_raw_reject(first_samp):
Expand Down

0 comments on commit 7100772

Please sign in to comment.