def test_assertion_dmindmax(self): # test digital and dmin wrong signals =[np.random.randint(-2048, 2048, 256*60).astype(np.int32)] sheaders = [highlevel.make_signal_header('ch1', sample_rate=256)] sheaders[0]['digital_min'] = -128 sheaders[0]['digital_max'] = 128 with self.assertRaises(AssertionError): highlevel.write_edf(self.edfplus_data_file, signals, sheaders, digital=True) # test pmin wrong signals = [np.random.randint(-2048, 2048, 256*60)] sheaders = [highlevel.make_signal_header('ch1', sample_rate=256)] sheaders[0]['physical_min'] = -200 sheaders[0]['physical_max'] = 200 with self.assertRaises(AssertionError): highlevel.write_edf(self.edfplus_data_file, signals, sheaders, digital=False)
def test_read_write_diff_sfreq(self): signals = [] sfreqs = [1, 64, 128, 200] sheaders = [] for sfreq in sfreqs: signals.append(np.random.randint(-2048, 2048, sfreq*60).astype(np.int32)) shead = highlevel.make_signal_header('ch{}'.format(sfreq), sample_rate=sfreq) sheaders.append(shead) highlevel.write_edf(self.edfplus_data_file, signals, sheaders, digital=True) signals2, sheaders2, _ = highlevel.read_edf(self.edfplus_data_file, digital=True) for s1, s2 in zip(signals, signals2): np.testing.assert_allclose(s1, s2)