def test_list_vs(bedmaster_reader: BedmasterReader, empty_matfile: h5py.File):
expected_vs = ["CO", "CUFF", "HR", "SPO2%", "SPO2R"]
assert bedmaster_reader.list_vs() == expected_vs
# No vs on file:
empty_reader = BedmasterReader(empty_matfile.name)
assert empty_reader.list_vs() == []
def test_list_wv(bedmaster_reader: BedmasterReader, empty_matfile: h5py.File):
expected_dict = {
"I": "ch7",
"II": "ch8",
"III": "ch9",
"V": "ch10",
"SPO2": "ch39",
"RESP": "ch40",
}
assert bedmaster_reader.list_wv() == expected_dict
# No wv on file:
empty_reader = BedmasterReader(empty_matfile.name)
assert empty_reader.list_vs() == []
def test_max_segment(bedmaster_reader: BedmasterReader):
def _create_max_seg_dict(seg_no, max_time, signal_name):
return {
"segmentNo": seg_no,
"maxTime": max_time,
"signalName": signal_name,
}
empty_vs_dict = _create_max_seg_dict(0, -1, "")
empty_wv_dict = _create_max_seg_dict(0, -1, "")
expected_wv_max = _create_max_seg_dict(
seg_no=6,
max_time=1452438403.75,
signal_name="ch10",
)
expected_vs_max = _create_max_seg_dict(
seg_no=6,
max_time=1452438402.0,
signal_name="CO",
)
assert bedmaster_reader.max_segment["vs"] == empty_vs_dict
assert bedmaster_reader.max_segment["wv"] == empty_wv_dict
wv_signals = bedmaster_reader.list_wv()
for wv_signal_name, channel in wv_signals.items():
bedmaster_reader.get_wv(channel, wv_signal_name)
assert bedmaster_reader.max_segment["vs"] == empty_vs_dict
assert bedmaster_reader.max_segment["wv"] == expected_wv_max
vs_signals = bedmaster_reader.list_vs()
for vs_signal_name in vs_signals:
bedmaster_reader.get_vs(vs_signal_name)
assert bedmaster_reader.max_segment["vs"] == expected_vs_max
assert bedmaster_reader.max_segment["wv"] == expected_wv_max