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)