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_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)
