def test_save_NS_efile_with_header(self):
fp = self.resdir + 'NS2_withheader.ev2'
fp2 = fp.replace('.ev2', '_recoded.ev2')
efile = load_efile(fp)
save_efile(efile)
efile2 = load_efile(fp2)
pattern = '[\s]+'
self.assertEqual(len(efile.raw), len(efile2.raw))
raw1 = [list(filter(bool, re.split(pattern, l))) for l in efile.raw]
raw2 = [list(filter(bool, re.split(pattern, l))) for l in efile2.raw]
self.assertEqual(raw1, raw2)
def test_save_BESA(self):
# BESA evt files seem to have spurious whitespace
fp = self.resdir + 'BESA_minimal.evt'
fp2 = fp.replace('.evt', '_recoded.evt')
efile = load_efile(fp)
save_efile(efile)
efile2 = load_efile(fp2)
pattern = '[\s,]+'
self.assertEqual(len(efile.raw), len(efile2.raw))
raw1 = [list(filter(bool, re.split(pattern, l))) for l in efile.raw]
raw2 = [list(filter(bool, re.split(pattern, l))) for l in efile2.raw]
self.assertEqual(raw1, raw2)
def test_save_Neuroscan(self):
fp = self.resdir + 'NS2_minimal.ev2'
fp2 = fp.replace('.ev2', '_recoded.ev2')
efile = load_efile(fp)
save_efile(efile)
self.assertTrue(filecmp.cmp(fp, fp2, shallow=False))
def test_mod_and_save(self):
efile = load_efile(self.NSfile)
efile.events[4].code = 247
save_efile(efile)
ef2 = load_efile(self.resdir + 'NS2_recoded.ev2')
self.assertEqual(ef2.events[4].code, 247)
