def test_eegplot_update():
print "testing update"
plot = EEGPlot('all')
plot.start()
plot.receive_band('theta1', '/muse/dsp/elements/theta', [0.1, 0.2, 0.3, 0.4])
print plot.data['theta1'][-1]
assert plot.data['theta1'][-1] == 0.2
def test_receive_fea():
plot = EEGPlot('fea')
plot.start()
plot.receive_fea('/muse/dsp/elements/alpha', [0.1, 0.2, 0.3, 0.4])
assert np.isclose(plot.data['fea'][-1],0.1)
plot.receive_fea('/muse/dsp/elements/alpha', [0.1, 0.3, 0.1, 0.4])
assert np.isclose(plot.data['fea'][-1],-0.2)
plot.receive_fea('/muse/dsp/elements/alpha', [0.1, 0.6, 0.6, 0.4])
assert np.isclose(plot.data['fea'][-1],0)
def test_save():
plot = EEGPlot('fea')
plot.start()
for i in range(10):
plot.receive_fea('/muse/dsp/elements/alpha', [0.1, np.random.random(), np.random.random(), 0.4])
saved_plot = plot.serialize()
assert saved_plot is not None
restored_plot = EEGPlot.deserialize(saved_plot)
assert np.allclose(restored_plot.data['fea'], plot.data['fea'])
def test_signal_descriptor():
assert EEGPlot.read_signal('all') is not None
assert EEGPlot.read_signal('frontal-alpha') == set(['alpha1', 'alpha2'])
assert EEGPlot.read_signal('front-delta') == set(['delta1', 'delta2'])
assert EEGPlot.read_signal('ears-beta') == set(['beta0', 'beta3'])
assert EEGPlot.read_signal('back-gamma') == set(['gamma0', 'gamma3'])
assert EEGPlot.read_signal('front-right-beta') == set(['beta2'])
assert EEGPlot.read_signal('front-left-alpha') == set(['alpha1'])
def parse_properties(self, props):
self.data_widgets = {}
props.setdefault('name', "")
props.setdefault('duration', "")
props.setdefault('next_button', "true")
props.setdefault('show_timer', "true")
props.setdefault('record', "true")
self.name = props['name']
self.record = props['record']
self.show_timer = (props['show_timer'] == 'true')
self.duration = self.parse_duration(props['duration'])
self.has_next_button = (props['next_button'] == 'true')
if 'graph' in props:
self.graph = True
plot_params = props['graph']
self.plot = EEGPlot(plot_params)
self.data_widgets['plot'] = self.plot
else:
self.graph = False
class Step(QObject):
def __init__(self, params):
super(Step, self).__init__()
self.parse_properties(params)
self.initUI()
print "Step {0} initialized".format(self.name)
self.data_dict = {"name": self.name}
def parse_properties(self, props):
self.data_widgets = {}
props.setdefault('name', "")
props.setdefault('duration', "")
props.setdefault('next_button', "true")
props.setdefault('show_timer', "true")
props.setdefault('record', "true")
self.name = props['name']
self.record = props['record']
self.show_timer = (props['show_timer'] == 'true')
self.duration = self.parse_duration(props['duration'])
self.has_next_button = (props['next_button'] == 'true')
if 'graph' in props:
self.graph = True
plot_params = props['graph']
self.plot = EEGPlot(plot_params)
self.data_widgets['plot'] = self.plot
else:
self.graph = False
def initUI(self):
self.widget = QWidget()
self.grid = QHBoxLayout()
if self.duration:
self.timer_widget = TimerWidget(self.duration, self.endStep)
if self.show_timer:
self.grid.addWidget(self.timer_widget)
if self.graph:
self.grid.addWidget(self.plot.pw)
self.horseshoe = HorseshoeWidget(self.plot.pw)
else:
self.horseshoe = HorseshoeWidget(self.widget)
self.widget.setLayout(self.grid)
def startStep(self, callback):
print "starting a step"
if self.duration:
self.timer_widget.start()
self.horseshoe.start()
if self.graph:
self.plot.start()
self.callback = callback
def endStep(self):
for i in range(self.grid.count()): self.grid.itemAt(i).widget().close() # clear the layout
if self.graph:
self.plot.stop()
if self.record:
for name, widget in self.data_widgets.iteritems():
self.data_dict[name] = widget.serialize()
if self.has_next_button:
self.next_button = QPushButton()
self.next_button.setText("Press to Continue")
self.next_button.clicked.connect(lambda: self.callback(self.data_dict))
self.grid.addWidget(self.next_button)
else:
self.callback(self.data_dict)
def parse_duration(self, time_string):
if len(time_string) == 0:
return None
else:
times = [int(x) for x in time_string.split(":")]
times.reverse() # Now we have ss:mm:hh:dd
multipliers = [1, 60, 3600, 3600*24]
time_in_secs = sum([unit* multiplier for unit, multiplier in zip(times, multipliers)])
return time_in_secs
def test_eegplot_setup():
plot = EEGPlot('all')
assert isinstance(plot.get_widget(), PlotWidget)
