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])))