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)