def test_write(self): file_in = os.path.join(test_files_dir, '100_CH9.continuous') file_out = os.path.join(test_files_dir, 'write_test') channel_data = loadContinuous(file_in, dtype=np.int16, trim_last_record=False) hs = get_header_string(file_in) write_continuous(file_out, [channel_data, ], hs) self.assertTrue(filecmp.cmp(file_in, file_out))
def test_concat(self): file_in = os.path.join(test_files_dir, '100_CH9.continuous') file_in_2 = os.path.join(test_files_dir, '100_CH9_2.continuous') file_out = os.path.join(test_files_dir, 'write_test') ch_data = loadContinuous(file_in, dtype=np.int16, trim_last_record=False) ch_data_2 = loadContinuous(file_in_2, dtype=np.int16, trim_last_record=False) hs = get_header_string(file_in) write_continuous(file_out, (ch_data, ch_data_2), hs) ch_data_out = loadContinuous(file_out, dtype=np.int16, trim_last_record=False) ae(np.concatenate((ch_data['timestamps'], ch_data_2['timestamps'])), ch_data_out['timestamps'])