def test_set_time(self):
'''
Call the set_time function to improve (functional) code coverage
'''
blackboard = Blackboard()
simulator = DataSimulator(blackboard)
simulator.set_time(5)
def test_set_time(self):
'''
Call the set_time function to improve (functional) code coverage
'''
blackboard = Blackboard()
simulator = DataSimulator( blackboard )
simulator.set_time(5)
def _test_get_time(self):
bb = Blackboard()
sim = DataSimulator(bb)
reader = HDFReader()
filename = filenames[5][0]
# Reading two datafiles
[mat, frequencies,
_] = reader.read(filename) # Read data from first data file
data = [np.asarray(mat[i - 1][1])
for i in range(1, 9)] # get data from selected channels
patterns = ['/EEG/channel_%d' % (i) for i in range(1, 9)]
interval = 1. / frequencies[1]
sim.set_simulation_data(data)
sim.set_simulation_interval(interval)
sim.set_patterns(patterns)
total_time = sim.get_total_time()
start_time = sim.get_current_time()
self.assertEquals(start_time, 0, 'Time should be zero before playback')
sim.start()
time.sleep(3)
now_time = sim.get_current_time()
self.assertAlmostEquals(
now_time,
3,
msg='Time should be close to 3 after 3 seconds playback. Is %f' %
now_time,
delta=.05)
sim.set_time(total_time - 2)
self.assertTrue(sim.get_simulation_state() is DataSimulator.PLAYING,
'Should be playing after setting time')
time.sleep(3)
self.assertTrue(sim.get_simulation_state() is DataSimulator.IDLE,
'Should be done playing')
sim.stop()
def _test_get_time(self):
bb = Blackboard()
sim = DataSimulator( bb )
reader = HDFReader();
filename = filenames[5][0]
# Reading two datafiles
[ mat, frequencies, _ ] = reader.read(filename) # Read data from first data file
data = [ np.asarray( mat[i-1][ 1 ] ) for i in range(1,9) ] # get data from selected channels
patterns = ['/EEG/channel_%d' % (i) for i in range(1,9) ]
interval = 1. / frequencies[ 1 ]
sim.set_simulation_data(data)
sim.set_simulation_interval(interval)
sim.set_patterns(patterns)
total_time = sim.get_total_time()
start_time = sim.get_current_time()
self.assertEquals(start_time, 0, 'Time should be zero before playback')
sim.start()
time.sleep( 3 )
now_time = sim.get_current_time()
self.assertAlmostEquals(now_time, 3, msg='Time should be close to 3 after 3 seconds playback. Is %f'%now_time, delta=.05)
sim.set_time( total_time - 2 )
self.assertTrue( sim.get_simulation_state() is DataSimulator.PLAYING, 'Should be playing after setting time')
time.sleep(3)
self.assertTrue( sim.get_simulation_state() is DataSimulator.IDLE, 'Should be done playing')
sim.stop()