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