def test2(self):
"""Checks that cluster id autosets to 0 without clu file"""
kio = KlustaKwikIO(filename=os.path.join(self.dirname, 'base2'),
sampling_rate=1000.)
block = kio.read()[0]
seg = block.segments[0]
self.assertEqual(len(seg.spiketrains), 1)
self.assertEqual(seg.spiketrains[0].name, 'unit 0 from group 5')
self.assertEqual(seg.spiketrains[0].annotations['cluster'], 0)
self.assertEqual(seg.spiketrains[0].annotations['group'], 5)
self.assertEqual(seg.spiketrains[0].t_start, 0.0)
self.assertTrue(np.all(seg.spiketrains[0].times == np.array(
[0.026, 0.122, 0.228])))
def test2(self):
"""Checks that cluster id autosets to 0 without clu file"""
kio = KlustaKwikIO(filename=os.path.join(self.dirname, 'base2'),
sampling_rate=1000.)
block = kio.read()
seg = block.segments[0]
self.assertEqual(len(seg.spiketrains), 1)
self.assertEqual(seg.spiketrains[0].name, 'unit 0 from group 5')
self.assertEqual(seg.spiketrains[0].annotations['cluster'], 0)
self.assertEqual(seg.spiketrains[0].annotations['group'], 5)
self.assertEqual(seg.spiketrains[0].t_start, 0.0)
self.assertTrue(np.all(seg.spiketrains[0].times == np.array(
[0.026, 0.122, 0.228])))
def test1(self):
"""Tests that data and metadata are read correctly"""
kio = KlustaKwikIO(filename=os.path.join(self.dirname, 'base'),
sampling_rate=1000.)
block = kio.read()[0]
seg = block.segments[0]
self.assertEqual(len(seg.spiketrains), 4)
for st in seg.spiketrains:
self.assertEqual(st.units, np.array(1.0) * pq.s)
self.assertEqual(st.t_start, 0.0)
self.assertEqual(seg.spiketrains[0].name, 'unit 1 from group 0')
self.assertEqual(seg.spiketrains[0].annotations['cluster'], 1)
self.assertEqual(seg.spiketrains[0].annotations['group'], 0)
self.assertTrue(np.all(seg.spiketrains[0].times == np.array(
[.100, .200])))
self.assertEqual(seg.spiketrains[1].name, 'unit 2 from group 0')
self.assertEqual(seg.spiketrains[1].annotations['cluster'], 2)
self.assertEqual(seg.spiketrains[1].annotations['group'], 0)
self.assertEqual(seg.spiketrains[1].t_start, 0.0)
self.assertTrue(np.all(seg.spiketrains[1].times == np.array([.305])))
self.assertEqual(seg.spiketrains[2].name, 'unit -1 from group 1')
self.assertEqual(seg.spiketrains[2].annotations['cluster'], -1)
self.assertEqual(seg.spiketrains[2].annotations['group'], 1)
self.assertEqual(seg.spiketrains[2].t_start, 0.0)
self.assertTrue(np.all(seg.spiketrains[2].times == np.array([.253])))
self.assertEqual(seg.spiketrains[3].name, 'unit 2 from group 1')
self.assertEqual(seg.spiketrains[3].annotations['cluster'], 2)
self.assertEqual(seg.spiketrains[3].annotations['group'], 1)
self.assertEqual(seg.spiketrains[3].t_start, 0.0)
self.assertTrue(np.all(seg.spiketrains[3].times == np.array(
[.050, .152])))
def test1(self):
"""Tests that data and metadata are read correctly"""
kio = KlustaKwikIO(filename=os.path.join(self.dirname, 'base'),
sampling_rate=1000.)
block = kio.read()
seg = block.segments[0]
self.assertEqual(len(seg.spiketrains), 4)
for st in seg.spiketrains:
self.assertEqual(st.units, np.array(1.0) * pq.s)
self.assertEqual(st.t_start, 0.0)
self.assertEqual(seg.spiketrains[0].name, 'unit 1 from group 0')
self.assertEqual(seg.spiketrains[0].annotations['cluster'], 1)
self.assertEqual(seg.spiketrains[0].annotations['group'], 0)
self.assertTrue(np.all(seg.spiketrains[0].times == np.array(
[.100, .200])))
self.assertEqual(seg.spiketrains[1].name, 'unit 2 from group 0')
self.assertEqual(seg.spiketrains[1].annotations['cluster'], 2)
self.assertEqual(seg.spiketrains[1].annotations['group'], 0)
self.assertEqual(seg.spiketrains[1].t_start, 0.0)
self.assertTrue(np.all(seg.spiketrains[1].times == np.array([.305])))
self.assertEqual(seg.spiketrains[2].name, 'unit -1 from group 1')
self.assertEqual(seg.spiketrains[2].annotations['cluster'], -1)
self.assertEqual(seg.spiketrains[2].annotations['group'], 1)
self.assertEqual(seg.spiketrains[2].t_start, 0.0)
self.assertTrue(np.all(seg.spiketrains[2].times == np.array([.253])))
self.assertEqual(seg.spiketrains[3].name, 'unit 2 from group 1')
self.assertEqual(seg.spiketrains[3].annotations['cluster'], 2)
self.assertEqual(seg.spiketrains[3].annotations['group'], 1)
self.assertEqual(seg.spiketrains[3].t_start, 0.0)
self.assertTrue(np.all(seg.spiketrains[3].times == np.array(
[.050, .152])))
