Ejemplo n.º 1
0
    def setup_units(self):
        params = {'testarg2': 'yes', 'testarg3': True}
        self.unit1 = Unit(name='test',
                          description='tester 1',
                          file_origin='test.file',
                          channels_indexes=[1],
                          testarg1=1,
                          **params)
        self.unit2 = Unit(name='test',
                          description='tester 2',
                          file_origin='test.file',
                          channels_indexes=[2],
                          testarg1=1,
                          **params)
        self.unit1.annotate(testarg1=1.1, testarg0=[1, 2, 3])
        self.unit2.annotate(testarg11=1.1, testarg10=[1, 2, 3])

        self.unit1.spiketrains = self.train1
        self.unit2.spiketrains = self.train2

        self.unit1.spikes = self.spike1
        self.unit2.spikes = self.spike2

        create_many_to_one_relationship(self.unit1)
        create_many_to_one_relationship(self.unit2)
Ejemplo n.º 2
0
    def setup_units(self):
        params = {'testarg2': 'yes', 'testarg3': True}
        self.unit1 = Unit(name='test', description='tester 1',
                          file_origin='test.file', channels_indexes=[1],
                          testarg1=1, **params)
        self.unit2 = Unit(name='test', description='tester 2',
                          file_origin='test.file', channels_indexes=[2],
                          testarg1=1, **params)
        self.unit1.annotate(testarg1=1.1, testarg0=[1, 2, 3])
        self.unit2.annotate(testarg11=1.1, testarg10=[1, 2, 3])

        self.unit1.spiketrains = self.train1
        self.unit2.spiketrains = self.train2

        self.unit1.spikes = self.spike1
        self.unit2.spikes = self.spike2

        self.unit1.create_many_to_one_relationship()
        self.unit2.create_many_to_one_relationship()
Ejemplo n.º 3
0
    def setup_unit(self):
        unitname11 = 'unit 1 1'
        unitname12 = 'unit 1 2'
        unitname21 = 'unit 2 1'
        unitname22 = 'unit 2 2'

        self.unitnames1 = [unitname11, unitname12]
        self.unitnames2 = [unitname21, unitname22, unitname11]
        self.unitnames = [unitname11, unitname12, unitname21, unitname22]

        unit11 = Unit(name=unitname11, channel_indexes=np.array([1]))
        unit12 = Unit(name=unitname12, channel_indexes=np.array([2]))
        unit21 = Unit(name=unitname21, channel_indexes=np.array([1]))
        unit22 = Unit(name=unitname22, channel_indexes=np.array([2]))
        unit23 = Unit(name=unitname11, channel_indexes=np.array([1]))

        self.units1 = [unit11, unit12]
        self.units2 = [unit21, unit22, unit23]
        self.units = [unit11, unit12, unit21, unit22]
Ejemplo n.º 4
0
    def setUp(self):
        unitname11 = 'unit 1 1'
        unitname12 = 'unit 1 2'
        unitname21 = 'unit 2 1'
        unitname22 = 'unit 2 2'

        channame11 = 'chan 1 1'
        channame12 = 'chan 1 2'
        channame21 = 'chan 2 1'
        channame22 = 'chan 2 2'

        segname11 = 'seg 1 1'
        segname12 = 'seg 1 2'
        segname21 = 'seg 2 1'
        segname22 = 'seg 2 2'

        self.rcgname1 = 'rcg 1'
        self.rcgname2 = 'rcg 2'
        self.rcgnames = [self.rcgname1, self.rcgname2]

        self.unitnames1 = [unitname11, unitname12]
        self.unitnames2 = [unitname21, unitname22, unitname11]
        self.unitnames = [unitname11, unitname12, unitname21, unitname22]

        self.channames1 = [channame11, channame12]
        self.channames2 = [channame21, channame22, channame11]
        self.channames = [channame11, channame12, channame21, channame22]

        self.segnames1 = [segname11, segname12]
        self.segnames2 = [segname21, segname22, segname11]
        self.segnames = [segname11, segname12, segname21, segname22]

        unit11 = Unit(name=unitname11)
        unit12 = Unit(name=unitname12)
        unit21 = Unit(name=unitname21)
        unit22 = Unit(name=unitname22)
        unit23 = unit11

        chan11 = RecordingChannel(name=channame11)
        chan12 = RecordingChannel(name=channame12)
        chan21 = RecordingChannel(name=channame21)
        chan22 = RecordingChannel(name=channame22)
        chan23 = chan11

        seg11 = Segment(name=segname11)
        seg12 = Segment(name=segname12)
        seg21 = Segment(name=segname21)
        seg22 = Segment(name=segname22)
        seg23 = seg11

        self.units1 = [unit11, unit12]
        self.units2 = [unit21, unit22, unit23]
        self.units = [unit11, unit12, unit21, unit22]

        self.chan1 = [chan11, chan12]
        self.chan2 = [chan21, chan22, chan23]
        self.chan = [chan11, chan12, chan21, chan22]

        self.seg1 = [seg11, seg12]
        self.seg2 = [seg21, seg22, seg23]
        self.seg = [seg11, seg12, seg21, seg22]

        self.rcg1 = RecordingChannelGroup(name=self.rcgname1)
        self.rcg2 = RecordingChannelGroup(name=self.rcgname2)
        self.rcg = [self.rcg1, self.rcg2]

        self.rcg1.units = self.units1
        self.rcg2.units = self.units2
        self.rcg1.recordingchannels = self.chan1
        self.rcg2.recordingchannels = self.chan2
Ejemplo n.º 5
0
class TestUnit(unittest.TestCase):
    def setUp(self):
        self.setup_spikes()
        self.setup_spiketrains()
        self.setup_units()

    def setup_units(self):
        params = {'testarg2': 'yes', 'testarg3': True}
        self.unit1 = Unit(name='test',
                          description='tester 1',
                          file_origin='test.file',
                          channels_indexes=[1],
                          testarg1=1,
                          **params)
        self.unit2 = Unit(name='test',
                          description='tester 2',
                          file_origin='test.file',
                          channels_indexes=[2],
                          testarg1=1,
                          **params)
        self.unit1.annotate(testarg1=1.1, testarg0=[1, 2, 3])
        self.unit2.annotate(testarg11=1.1, testarg10=[1, 2, 3])

        self.unit1.spiketrains = self.train1
        self.unit2.spiketrains = self.train2

        self.unit1.spikes = self.spike1
        self.unit2.spikes = self.spike2

        create_many_to_one_relationship(self.unit1)
        create_many_to_one_relationship(self.unit2)

    def setup_spikes(self):
        spikename11 = 'spike 1 1'
        spikename12 = 'spike 1 2'
        spikename21 = 'spike 2 1'
        spikename22 = 'spike 2 2'

        spikedata11 = 10 * pq.ms
        spikedata12 = 20 * pq.ms
        spikedata21 = 30 * pq.s
        spikedata22 = 40 * pq.s

        self.spikenames1 = [spikename11, spikename12]
        self.spikenames2 = [spikename21, spikename22]
        self.spikenames = [spikename11, spikename12, spikename21, spikename22]

        spike11 = Spike(spikedata11, t_stop=100 * pq.s, name=spikename11)
        spike12 = Spike(spikedata12, t_stop=100 * pq.s, name=spikename12)
        spike21 = Spike(spikedata21, t_stop=100 * pq.s, name=spikename21)
        spike22 = Spike(spikedata22, t_stop=100 * pq.s, name=spikename22)

        self.spike1 = [spike11, spike12]
        self.spike2 = [spike21, spike22]
        self.spike = [spike11, spike12, spike21, spike22]

    def setup_spiketrains(self):
        trainname11 = 'spiketrain 1 1'
        trainname12 = 'spiketrain 1 2'
        trainname21 = 'spiketrain 2 1'
        trainname22 = 'spiketrain 2 2'

        traindata11 = np.arange(0, 10) * pq.ms
        traindata12 = np.arange(10, 20) * pq.ms
        traindata21 = np.arange(20, 30) * pq.s
        traindata22 = np.arange(30, 40) * pq.s

        self.trainnames1 = [trainname11, trainname12]
        self.trainnames2 = [trainname21, trainname22]
        self.trainnames = [trainname11, trainname12, trainname21, trainname22]

        train11 = SpikeTrain(traindata11, t_stop=100 * pq.s, name=trainname11)
        train12 = SpikeTrain(traindata12, t_stop=100 * pq.s, name=trainname12)
        train21 = SpikeTrain(traindata21, t_stop=100 * pq.s, name=trainname21)
        train22 = SpikeTrain(traindata22, t_stop=100 * pq.s, name=trainname22)

        self.train1 = [train11, train12]
        self.train2 = [train21, train22]
        self.train = [train11, train12, train21, train22]

    def test_unit_creation(self):
        assert_neo_object_is_compliant(self.unit1)
        assert_neo_object_is_compliant(self.unit2)

        self.assertEqual(self.unit1.name, 'test')
        self.assertEqual(self.unit2.name, 'test')

        self.assertEqual(self.unit1.description, 'tester 1')
        self.assertEqual(self.unit2.description, 'tester 2')

        self.assertEqual(self.unit1.file_origin, 'test.file')
        self.assertEqual(self.unit2.file_origin, 'test.file')

        self.assertEqual(self.unit1.annotations['testarg0'], [1, 2, 3])
        self.assertEqual(self.unit2.annotations['testarg10'], [1, 2, 3])

        self.assertEqual(self.unit1.annotations['testarg1'], 1.1)
        self.assertEqual(self.unit2.annotations['testarg1'], 1)
        self.assertEqual(self.unit2.annotations['testarg11'], 1.1)

        self.assertEqual(self.unit1.annotations['testarg2'], 'yes')
        self.assertEqual(self.unit2.annotations['testarg2'], 'yes')

        self.assertTrue(self.unit1.annotations['testarg3'])
        self.assertTrue(self.unit2.annotations['testarg3'])

        self.assertTrue(hasattr(self.unit1, 'spikes'))
        self.assertTrue(hasattr(self.unit2, 'spikes'))

        self.assertEqual(len(self.unit1.spikes), 2)
        self.assertEqual(len(self.unit2.spikes), 2)

        for res, targ in zip(self.unit1.spikes, self.spike1):
            self.assertEqual(res, targ)
            self.assertEqual(res.name, targ.name)

        for res, targ in zip(self.unit2.spikes, self.spike2):
            self.assertEqual(res, targ)
            self.assertEqual(res.name, targ.name)

        self.assertTrue(hasattr(self.unit1, 'spiketrains'))
        self.assertTrue(hasattr(self.unit2, 'spiketrains'))

        self.assertEqual(len(self.unit1.spiketrains), 2)
        self.assertEqual(len(self.unit2.spiketrains), 2)

        for res, targ in zip(self.unit1.spiketrains, self.train1):
            assert_arrays_equal(res, targ)
            self.assertEqual(res.name, targ.name)

        for res, targ in zip(self.unit2.spiketrains, self.train2):
            assert_arrays_equal(res, targ)
            self.assertEqual(res.name, targ.name)

    def test_unit_merge(self):
        self.unit1.merge(self.unit2)

        spikeres1 = [sig.name for sig in self.unit1.spikes]
        spikeres2 = [sig.name for sig in self.unit2.spikes]

        trainres1 = [sig.name for sig in self.unit1.spiketrains]
        trainres2 = [sig.name for sig in self.unit2.spiketrains]

        self.assertEqual(spikeres1, self.spikenames)
        self.assertEqual(spikeres2, self.spikenames2)

        self.assertEqual(trainres1, self.trainnames)
        self.assertEqual(trainres2, self.trainnames2)

        for res, targ in zip(self.unit1.spikes, self.spike):
            self.assertEqual(res, targ)

        for res, targ in zip(self.unit2.spikes, self.spike2):
            self.assertEqual(res, targ)

        for res, targ in zip(self.unit1.spiketrains, self.train):
            assert_arrays_equal(res, targ)

        for res, targ in zip(self.unit2.spiketrains, self.train2):
            assert_arrays_equal(res, targ)
Ejemplo n.º 6
0
class TestUnit(unittest.TestCase):
    def setUp(self):
        self.setup_spikes()
        self.setup_spiketrains()
        self.setup_units()

    def setup_units(self):
        params = {'testarg2': 'yes', 'testarg3': True}
        self.unit1 = Unit(name='test', description='tester 1',
                          file_origin='test.file', channels_indexes=[1],
                          testarg1=1, **params)
        self.unit2 = Unit(name='test', description='tester 2',
                          file_origin='test.file', channels_indexes=[2],
                          testarg1=1, **params)
        self.unit1.annotate(testarg1=1.1, testarg0=[1, 2, 3])
        self.unit2.annotate(testarg11=1.1, testarg10=[1, 2, 3])

        self.unit1.spiketrains = self.train1
        self.unit2.spiketrains = self.train2

        self.unit1.spikes = self.spike1
        self.unit2.spikes = self.spike2

        self.unit1.create_many_to_one_relationship()
        self.unit2.create_many_to_one_relationship()

    def setup_spikes(self):
        spikename11 = 'spike 1 1'
        spikename12 = 'spike 1 2'
        spikename21 = 'spike 2 1'
        spikename22 = 'spike 2 2'

        spikedata11 = 10 * pq.ms
        spikedata12 = 20 * pq.ms
        spikedata21 = 30 * pq.s
        spikedata22 = 40 * pq.s

        self.spikenames1 = [spikename11, spikename12]
        self.spikenames2 = [spikename21, spikename22]
        self.spikenames = [spikename11, spikename12, spikename21, spikename22]

        spike11 = Spike(spikedata11, t_stop=100*pq.s, name=spikename11)
        spike12 = Spike(spikedata12, t_stop=100*pq.s, name=spikename12)
        spike21 = Spike(spikedata21, t_stop=100*pq.s, name=spikename21)
        spike22 = Spike(spikedata22, t_stop=100*pq.s, name=spikename22)

        self.spike1 = [spike11, spike12]
        self.spike2 = [spike21, spike22]
        self.spike = [spike11, spike12, spike21, spike22]

    def setup_spiketrains(self):
        trainname11 = 'spiketrain 1 1'
        trainname12 = 'spiketrain 1 2'
        trainname21 = 'spiketrain 2 1'
        trainname22 = 'spiketrain 2 2'

        traindata11 = np.arange(0, 10) * pq.ms
        traindata12 = np.arange(10, 20) * pq.ms
        traindata21 = np.arange(20, 30) * pq.s
        traindata22 = np.arange(30, 40) * pq.s

        self.trainnames1 = [trainname11, trainname12]
        self.trainnames2 = [trainname21, trainname22]
        self.trainnames = [trainname11, trainname12, trainname21, trainname22]

        train11 = SpikeTrain(traindata11, t_stop=100*pq.s, name=trainname11)
        train12 = SpikeTrain(traindata12, t_stop=100*pq.s, name=trainname12)
        train21 = SpikeTrain(traindata21, t_stop=100*pq.s, name=trainname21)
        train22 = SpikeTrain(traindata22, t_stop=100*pq.s, name=trainname22)

        self.train1 = [train11, train12]
        self.train2 = [train21, train22]
        self.train = [train11, train12, train21, train22]

    def test_unit_creation(self):
        assert_neo_object_is_compliant(self.unit1)
        assert_neo_object_is_compliant(self.unit2)

        self.assertEqual(self.unit1.name, 'test')
        self.assertEqual(self.unit2.name, 'test')

        self.assertEqual(self.unit1.description, 'tester 1')
        self.assertEqual(self.unit2.description, 'tester 2')

        self.assertEqual(self.unit1.file_origin, 'test.file')
        self.assertEqual(self.unit2.file_origin, 'test.file')

        self.assertEqual(self.unit1.annotations['testarg0'], [1, 2, 3])
        self.assertEqual(self.unit2.annotations['testarg10'], [1, 2, 3])

        self.assertEqual(self.unit1.annotations['testarg1'], 1.1)
        self.assertEqual(self.unit2.annotations['testarg1'], 1)
        self.assertEqual(self.unit2.annotations['testarg11'], 1.1)

        self.assertEqual(self.unit1.annotations['testarg2'], 'yes')
        self.assertEqual(self.unit2.annotations['testarg2'], 'yes')

        self.assertTrue(self.unit1.annotations['testarg3'])
        self.assertTrue(self.unit2.annotations['testarg3'])

        self.assertTrue(hasattr(self.unit1, 'spikes'))
        self.assertTrue(hasattr(self.unit2, 'spikes'))

        self.assertEqual(len(self.unit1.spikes), 2)
        self.assertEqual(len(self.unit2.spikes), 2)

        for res, targ in zip(self.unit1.spikes, self.spike1):
            self.assertEqual(res, targ)
            self.assertEqual(res.name, targ.name)

        for res, targ in zip(self.unit2.spikes, self.spike2):
            self.assertEqual(res, targ)
            self.assertEqual(res.name, targ.name)

        self.assertTrue(hasattr(self.unit1, 'spiketrains'))
        self.assertTrue(hasattr(self.unit2, 'spiketrains'))

        self.assertEqual(len(self.unit1.spiketrains), 2)
        self.assertEqual(len(self.unit2.spiketrains), 2)

        for res, targ in zip(self.unit1.spiketrains, self.train1):
            assert_arrays_equal(res, targ)
            self.assertEqual(res.name, targ.name)

        for res, targ in zip(self.unit2.spiketrains, self.train2):
            assert_arrays_equal(res, targ)
            self.assertEqual(res.name, targ.name)

    def test_unit_merge(self):
        self.unit1.merge(self.unit2)

        spikeres1 = [sig.name for sig in self.unit1.spikes]
        spikeres2 = [sig.name for sig in self.unit2.spikes]

        trainres1 = [sig.name for sig in self.unit1.spiketrains]
        trainres2 = [sig.name for sig in self.unit2.spiketrains]

        self.assertEqual(spikeres1, self.spikenames)
        self.assertEqual(spikeres2, self.spikenames2)

        self.assertEqual(trainres1, self.trainnames)
        self.assertEqual(trainres2, self.trainnames2)

        for res, targ in zip(self.unit1.spikes, self.spike):
            self.assertEqual(res, targ)

        for res, targ in zip(self.unit2.spikes, self.spike2):
            self.assertEqual(res, targ)

        for res, targ in zip(self.unit1.spiketrains, self.train):
            assert_arrays_equal(res, targ)

        for res, targ in zip(self.unit2.spiketrains, self.train2):
            assert_arrays_equal(res, targ)

    def test__children(self):
        rcg = RecordingChannelGroup(name='rcg1')
        rcg.units = [self.unit1]
        rcg.create_many_to_one_relationship()

        self.assertEqual(self.unit1._container_child_objects, ())
        self.assertEqual(self.unit1._data_child_objects,
                         ('Spike', 'SpikeTrain'))
        self.assertEqual(self.unit1._single_parent_objects,
                         ('RecordingChannelGroup',))
        self.assertEqual(self.unit1._multi_child_objects, ())
        self.assertEqual(self.unit1._multi_parent_objects, ())
        self.assertEqual(self.unit1._child_properties, ())

        self.assertEqual(self.unit1._single_child_objects,
                         ('Spike', 'SpikeTrain'))

        self.assertEqual(self.unit1._container_child_containers, ())
        self.assertEqual(self.unit1._data_child_containers,
                         ('spikes', 'spiketrains'))
        self.assertEqual(self.unit1._single_child_containers,
                         ('spikes', 'spiketrains'))
        self.assertEqual(self.unit1._single_parent_containers,
                         ('recordingchannelgroup',))
        self.assertEqual(self.unit1._multi_child_containers, ())
        self.assertEqual(self.unit1._multi_parent_containers, ())

        self.assertEqual(self.unit1._child_objects,
                         ('Spike', 'SpikeTrain'))
        self.assertEqual(self.unit1._child_containers,
                         ('spikes', 'spiketrains'))
        self.assertEqual(self.unit1._parent_objects,
                         ('RecordingChannelGroup',))
        self.assertEqual(self.unit1._parent_containers,
                         ('recordingchannelgroup',))

        self.assertEqual(len(self.unit1.children),
                         len(self.spike1) + len(self.train1))
        self.assertEqual(self.unit1.children[0].name, self.spikenames1[0])
        self.assertEqual(self.unit1.children[1].name, self.spikenames1[1])
        self.assertEqual(self.unit1.children[2].name, self.trainnames1[0])
        self.assertEqual(self.unit1.children[3].name, self.trainnames1[1])
        self.assertEqual(len(self.unit1.parents), 1)
        self.assertEqual(self.unit1.parents[0].name, 'rcg1')

        self.unit1.create_many_to_one_relationship()
        assert_neo_object_is_compliant(self.unit1)