def test__children(self): params = {'test2': 'y1', 'test3': True} evta = EventArray([1.1, 1.5, 1.7] * pq.ms, labels=np.array( ['test event 1', 'test event 2', 'test event 3'], dtype='S'), name='test', description='tester', file_origin='test.file', test1=1, **params) evta.annotate(test1=1.1, test0=[1, 2]) assert_neo_object_is_compliant(evta) segment = Segment(name='seg1') segment.eventarrays = [evta] segment.create_many_to_one_relationship() self.assertEqual(evta._single_parent_objects, ('Segment', )) self.assertEqual(evta._multi_parent_objects, ()) self.assertEqual(evta._single_parent_containers, ('segment', )) self.assertEqual(evta._multi_parent_containers, ()) self.assertEqual(evta._parent_objects, ('Segment', )) self.assertEqual(evta._parent_containers, ('segment', )) self.assertEqual(len(evta.parents), 1) self.assertEqual(evta.parents[0].name, 'seg1') assert_neo_object_is_compliant(evta)
def test__children(self): params = {'test2': 'y1', 'test3': True} evta = EventArray([1.1, 1.5, 1.7]*pq.ms, labels=np.array(['test event 1', 'test event 2', 'test event 3'], dtype='S'), name='test', description='tester', file_origin='test.file', test1=1, **params) evta.annotate(test1=1.1, test0=[1, 2]) assert_neo_object_is_compliant(evta) segment = Segment(name='seg1') segment.eventarrays = [evta] segment.create_many_to_one_relationship() self.assertEqual(evta._single_parent_objects, ('Segment',)) self.assertEqual(evta._multi_parent_objects, ()) self.assertEqual(evta._single_parent_containers, ('segment',)) self.assertEqual(evta._multi_parent_containers, ()) self.assertEqual(evta._parent_objects, ('Segment',)) self.assertEqual(evta._parent_containers, ('segment',)) self.assertEqual(len(evta.parents), 1) self.assertEqual(evta.parents[0].name, 'seg1') assert_neo_object_is_compliant(evta)
def proc_src_comments(srcfile, filename): '''Get the comments in an src file that has been#!N processed by the official matlab function. See proc_src for details''' comm_seg = Segment(name='Comments', file_origin=filename) commentarray = srcfile['comments'].flatten()[0] senders = [res[0] for res in commentarray['sender'].flatten()] texts = [res[0] for res in commentarray['text'].flatten()] timeStamps = [res[0, 0] for res in commentarray['timeStamp'].flatten()] timeStamps = np.array(timeStamps, dtype=np.float32) t_start = timeStamps.min() timeStamps = pq.Quantity(timeStamps - t_start, units=pq.d).rescale(pq.s) texts = np.array(texts, dtype='S') senders = np.array(senders, dtype='S') t_start = brainwaresrcio.convert_brainwaresrc_timestamp(t_start.tolist()) comments = EventArray(times=timeStamps, labels=texts, senders=senders) comm_seg.eventarrays = [comments] comm_seg.rec_datetime = t_start return comm_seg
def proc_src_comments(srcfile, filename): '''Get the comments in an src file that has been#!N processed by the official matlab function. See proc_src for details''' comm_seg = Segment(name='Comments', file_origin=filename) commentarray = srcfile['comments'].flatten()[0] senders = [res[0] for res in commentarray['sender'].flatten()] texts = [res[0] for res in commentarray['text'].flatten()] timeStamps = [res[0, 0] for res in commentarray['timeStamp'].flatten()] timeStamps = np.array(timeStamps, dtype=np.float32) t_start = timeStamps.min() timeStamps = pq.Quantity(timeStamps-t_start, units=pq.d).rescale(pq.s) texts = np.array(texts, dtype='S') senders = np.array(senders, dtype='S') t_start = brainwaresrcio.convert_brainwaresrc_timestamp(t_start.tolist()) comments = EventArray(times=timeStamps, labels=texts, senders=senders) comm_seg.eventarrays = [comments] comm_seg.rec_datetime = t_start return comm_seg