def test_headset_4(self): """ Test recording state after streaming start """ record_time = 10 model = ApplicationModel() model.start_headsets() model.start_recording(record_time) model.stop_recording() model.stop_headsets() self.assertFalse(model.is_recording(), "Should not be recording")
class TestApplicationModelSimulation( unittest.TestCase ): def setUp(self): unittest.TestCase.setUp(self) self.model = ApplicationModel() _eeg_type = 1 [ mat, frequencies ] = HDFReader().read( filenames[6][0] ) data = [ np.asarray( mat[i-1][ _eeg_type ] ) for i in range(1,9) ] interval = 1. / frequencies[_eeg_type] self.model.simulators[0].set_simulation_data(data) self.model.simulators[0].set_simulation_interval(interval) eeg_chans = ['/eega0/channel_%d/pp' % i for i in range(1,9) ] self.model.simulators[0].set_patterns( eeg_chans ) def _test_modelstate_1(self): self.model.start_simulation() self.assertEquals( self.model.get_simulation_state(), DataSimulator.PLAYING ) self.model.stop_simulation() def _test_modelstate_2(self): self.model.start_simulation() self.model.stop_simulation() self.assertEquals( self.model.get_simulation_state(), DataSimulator.IDLE ) def _test_modelstate_3(self): self.model.start_simulation() self.model.pause_simulation() sleep( .2 ) self.assertEquals( self.model.get_simulation_state(), DataSimulator.PAUSED ) self.model.stop_simulation() def _test_modelstate_4(self): self.model.pause_simulation() self.assertEquals( self.model.get_simulation_state(), DataSimulator.IDLE ) def tearDown(self): unittest.TestCase.tearDown(self) self.model.stop_simulation() self.model.stop_recording() self.model.stop_headsets()