def test__get_epochs(self):
a = Epoch([0.5, 10.0, 25.2] * pq.s, durations=[5.1, 4.8, 5.0] * pq.s)
a.annotate(epoch_type='a', pick='me')
a.array_annotate(trial_id=[1, 2, 3])
b = Epoch([5.5, 14.9, 30.1] * pq.s, durations=[4.7, 4.9, 5.2] * pq.s)
b.annotate(epoch_type='b')
b.array_annotate(trial_id=[1, 2, 3])
proxy_epoch = EpochProxy(rawio=self.reader,
event_channel_index=1,
block_index=0,
seg_index=0)
proxy_epoch.annotate(epoch_type='a')
seg = Segment()
seg.epochs = [a, b, proxy_epoch]
# test getting multiple epochs including a proxy
extracted_epochs = get_epochs(seg, epoch_type='a')
self.assertEqual(len(extracted_epochs), 2)
# make sure the epoch is loaded and a neo.Epoch object is returned
self.assertTrue(isinstance(extracted_epochs[0], Epoch))
self.assertTrue(isinstance(extracted_epochs[1], Epoch))
def test__get_epochs(self):
a_1 = Epoch([0.5, 10.0, 25.2] * pq.s, durations=[5.1, 4.8, 5.0] * pq.s)
a_1.annotate(epoch_type='a', pick='me')
a_1.array_annotate(trial_id=[1, 2, 3])
b_1 = Epoch([5.5, 14.9, 30.1] * pq.s, durations=[4.7, 4.9, 5.2] * pq.s)
b_1.annotate(epoch_type='b')
b_1.array_annotate(trial_id=[1, 2, 3])
a_2 = Epoch([33.2, 41.7, 52.4] * pq.s,
durations=[5.3, 5.0, 5.1] * pq.s)
a_2.annotate(epoch_type='a')
a_2.array_annotate(trial_id=[4, 5, 6])
b_2 = Epoch([37.6, 46.1, 57.0] * pq.s,
durations=[4.9, 5.2, 5.1] * pq.s)
b_2.annotate(epoch_type='b')
b_2.array_annotate(trial_id=[4, 5, 6])
seg = Segment()
seg2 = Segment()
seg.epochs = [a_1, b_1]
seg2.epochs = [a_2, b_2]
block = Block()
block.segments = [seg, seg2]
# test getting one whole event via annotation
extracted_a_1 = get_epochs(seg, epoch_type='a')
extracted_a_1b = get_epochs(block, pick='me')
self.assertEqual(len(extracted_a_1), 1)
self.assertEqual(len(extracted_a_1b), 1)
extracted_a_1 = extracted_a_1[0]
extracted_a_1b = extracted_a_1b[0]
assert_same_attributes(extracted_a_1, a_1)
assert_same_attributes(extracted_a_1b, a_1)
# test getting an empty list by searching for a non-existent property
empty1 = get_epochs(seg, foo='bar')
self.assertEqual(len(empty1), 0)
# test getting an empty list by searching for a non-existent property value
empty2 = get_epochs(seg, epoch_type='undefined')
self.assertEqual(len(empty2), 0)
# test getting only one event time of one event
trial_2 = get_epochs(block, trial_id=2, epoch_type='a')
self.assertEqual(len(trial_2), 1)
trial_2 = trial_2[0]
self.assertEqual(a_1.name, trial_2.name)
self.assertEqual(a_1.description, trial_2.description)
self.assertEqual(a_1.file_origin, trial_2.file_origin)
self.assertEqual(a_1.annotations['epoch_type'],
trial_2.annotations['epoch_type'])
assert_arrays_equal(trial_2.array_annotations['trial_id'],
np.array([2]))
self.assertIsInstance(trial_2.array_annotations, ArrayDict)
# test getting only one event time of more than one event
trial_2b = get_epochs(block, trial_id=2)
self.assertEqual(len(trial_2b), 2)
a_idx = np.where(
np.array([ev.annotations['epoch_type']
for ev in trial_2b]) == 'a')[0][0]
trial_2b_a = trial_2b[a_idx]
trial_2b_b = trial_2b[a_idx - 1]
assert_same_attributes(trial_2b_a, trial_2)
self.assertEqual(b_1.name, trial_2b_b.name)
self.assertEqual(b_1.description, trial_2b_b.description)
self.assertEqual(b_1.file_origin, trial_2b_b.file_origin)
self.assertEqual(b_1.annotations['epoch_type'],
trial_2b_b.annotations['epoch_type'])
assert_arrays_equal(trial_2b_b.array_annotations['trial_id'],
np.array([2]))
self.assertIsInstance(trial_2b_b.array_annotations, ArrayDict)
# test getting more than one event time of one event
trials_1_2 = get_epochs(block, trial_id=[1, 2], epoch_type='a')
self.assertEqual(len(trials_1_2), 1)
trials_1_2 = trials_1_2[0]
self.assertEqual(a_1.name, trials_1_2.name)
self.assertEqual(a_1.description, trials_1_2.description)
self.assertEqual(a_1.file_origin, trials_1_2.file_origin)
self.assertEqual(a_1.annotations['epoch_type'],
trials_1_2.annotations['epoch_type'])
assert_arrays_equal(trials_1_2.array_annotations['trial_id'],
np.array([1, 2]))
self.assertIsInstance(trials_1_2.array_annotations, ArrayDict)
# test getting more than one event time of more than one event
trials_1_2b = get_epochs(block, trial_id=[1, 2])
self.assertEqual(len(trials_1_2b), 2)
a_idx = np.where(
np.array([ev.annotations['epoch_type']
for ev in trials_1_2b]) == 'a')[0][0]
trials_1_2b_a = trials_1_2b[a_idx]
trials_1_2b_b = trials_1_2b[a_idx - 1]
assert_same_attributes(trials_1_2b_a, trials_1_2)
self.assertEqual(b_1.name, trials_1_2b_b.name)
self.assertEqual(b_1.description, trials_1_2b_b.description)
self.assertEqual(b_1.file_origin, trials_1_2b_b.file_origin)
self.assertEqual(b_1.annotations['epoch_type'],
trials_1_2b_b.annotations['epoch_type'])
assert_arrays_equal(trials_1_2b_b.array_annotations['trial_id'],
np.array([1, 2]))
self.assertIsInstance(trials_1_2b_b.array_annotations, ArrayDict)