import numpy as np
import pylab as pl
from ptsa.data import ArrayWrapper, Events
from ptsa.wavelet import phase_pow_multi
# some general info
nchan = 2
samplerate = 200
nsamples = samplerate * 100
event_dur = samplerate * 1
buf_dur = 1.0
# generate fake data
dat = np.random.rand(nchan, nsamples)
aw = ArrayWrapper(dat, samplerate)
# generate fake events
eoffset = np.arange(event_dur * 2, nsamples - (2 * event_dur), event_dur)
esrc = [aw] * len(eoffset)
nrec = len(eoffset) / 2
recalled = [True] * nrec + [False] * (len(eoffset) - nrec)
events = Events(
np.rec.fromarrays([esrc, eoffset, recalled],
names='esrc,eoffset,recalled'))
# load in data with events (filter at the same time)
rdat = events[events.recalled == True].get_data(
0, # channel
1.0, # duration in sec
0.0, # offset in sec
def setUp(self):
self.dat = np.random.rand(10, 1000)
self.aw = ArrayWrapper(self.dat, 200)
self.eoffsets = [80, 140, 270]
