Skip to content

Commit

Permalink
test_record: test round-trip write/read of supported binary formats.
Browse files Browse the repository at this point in the history
We want to test that both reading and writing work correctly for all
supported formats (although currently some binary formats are
completely unsupported for writing.)

Previous test cases covered formats 80, 212, 16, and 24 (and the FLAC
formats) but did not cover format 32.
  • Loading branch information
Benjamin Moody committed Oct 16, 2024
1 parent 5427181 commit 2a9feb5
Showing 1 changed file with 21 additions and 0 deletions.
21 changes: 21 additions & 0 deletions tests/test_record.py
Original file line number Diff line number Diff line change
Expand Up @@ -227,6 +227,27 @@ def test_1f(self):
"Mismatch in %s" % name,
)

# Test writing all supported formats. (Currently not all signal
# formats are supported for output; keep this list in sync with
# 'wr_dat_file' in wfdb/io/_signal.py.)
OUTPUT_FMTS = ["80", "212", "16", "24", "32"]
channels = []
for i, fmt in enumerate(record.fmt):
if fmt in OUTPUT_FMTS:
channels.append(i)

partial_record = wfdb.rdrecord(
"sample-data/binformats",
physical=False,
channels=channels,
)
partial_record.wrsamp(write_dir=self.temp_path)
converted_record = wfdb.rdrecord(
os.path.join(self.temp_path, "binformats"),
physical=False,
)
assert partial_record == converted_record

def test_read_write_flac(self):
"""
All FLAC formats, multiple signal files in one record.
Expand Down

0 comments on commit 2a9feb5

Please sign in to comment.