def test_read_write_wg1_edf():
"""Test ability to open a test edf file"""
from electrolib.io.wg1 import Wg1File
from electrolib.io.edf import EDFFile
# Test conversion to edf
eegin = Wg1File(os.path.join('data', 'wg1.wg1'), 'r')
eegout = EDFFile(os.path.join('data', 'wg1_out.edf'), 'w')
eegout.write_from(eegin)
def test_read_head():
"""Test reading of the GDF header"""
from electrolib.io.gdf import GDFFile
from electrolib.io.edf import EDFFile
gdfin = GDFFile(os.path.join('data', 'nblocks.gdf'), 'r')
edfin = EDFFile(os.path.join('data', 'nblocks.edf'), 'r')
ghead = gdfin.get_head()
ehead = edfin.get_head()
assert ghead.tstart == ehead.tstart
assert ghead.nchan == ehead.nchan
assert ghead.nblocks == ehead.nblocks
gchan = gdfin.get_chan()
echan = edfin.get_chan()
assert len(gchan) == len(echan) == ehead.nchan
for gc, ec in zip(gchan, echan):
assert gc.label == ec.label
assert gc.dim == ec.dim
assert gc.calib == ec.calib
assert gc.offset == ec.offset
assert gc.nsblock == ec.nsblock
assert gc.prefilt == ec.prefilt
assert gc.transducer == ec.transducer
def test_read_write_edf():
"""Test ability to open a test edf file"""
from electrolib.io.edf import EDFFile
eegin = EDFFile(os.path.join('data', 'sin.edf'), 'r')
eegout = EDFFile(os.path.join('data', 'sin_out.edf'), 'w')
eegout.write_from(eegin)
