def make_fake_event_source():
all_events = []
for c in range(3):
ev_times = np.arange(0, 10., .5) + c*3
ev_labels = np.array(['Event{} num {}'.format(c, i) for i in range(ev_times.size)], dtype='U')
all_events.append({ 'time':ev_times, 'label':ev_labels, 'name':'Event{}'.format(c) })
source = ephyviewer.InMemoryEventSource(all_events=all_events)
return source
def test_InMemoryEventSource():
ev_times = np.arange(0, 10., .5)
ev_labels = np.array(
['Event0 num {}'.format(i) for i in range(ev_times.size)], dtype='U')
event0 = {'time': ev_times, 'label': ev_labels, 'name': 'Event0'}
ev_times = np.arange(-6, 8., 2.)
ev_labels = np.array(
['Event1 num {}'.format(i) for i in range(ev_times.size)], dtype='U')
event1 = {'time': ev_times, 'label': ev_labels, 'name': 'Event1'}
all_events = [event0, event1]
source = ephyviewer.InMemoryEventSource(all_events=all_events)
assert source.t_start == -6.
assert source.t_stop == 9.5
assert source.get_size(0) == 20