def test_spikes(self):
trains = {}
trains[0] = [neo.Spike(0 * pq.ms, waveform=[[-1, -2], [1, 2]] * pq.mV),
neo.Spike(0 * pq.ms, waveform=[[-1, -2], [1, 2]] * pq.mV)]
trains[1] = trains[0]
means = {}
means[0] = neo.Spike(0 * pq.ms, waveform=[[-1, -1], [1, 1]] * pq.mV)
exp = qa.variance_explained(trains, means)
self.assertAlmostEqual(exp[1][0], 1)
self.assertAlmostEqual(exp[1][0], 1)
self.assertAlmostEqual(exp[0][0], 1)
self.assertAlmostEqual(exp[0][1], 0.75)
def test_trains(self):
trains = {}
trains[0] = neo.SpikeTrain(
sp.zeros(2) * pq.ms, 0 * pq.ms,
waveforms=[[[-1, -2], [1, 2]], [[-1, -2], [1, 2]]] * pq.mV)
trains[1] = trains[0]
means = {}
means[1] = neo.Spike(0 * pq.ms, waveform=[[-1, -1], [1, 1]] * pq.mV)
exp = qa.variance_explained(trains, means)
self.assertAlmostEqual(exp[0][0], 1)
self.assertAlmostEqual(exp[0][0], 1)
self.assertAlmostEqual(exp[1][0], 1)
self.assertAlmostEqual(exp[1][1], 0.75)
def test_spikes(self):
trains = {}
trains[0] = [
neo.Spike(0 * pq.ms, waveform=[[-1, -2], [1, 2]] * pq.mV),
neo.Spike(0 * pq.ms, waveform=[[-1, -2], [1, 2]] * pq.mV)
]
trains[1] = trains[0]
means = {}
means[0] = neo.Spike(0 * pq.ms, waveform=[[-1, -1], [1, 1]] * pq.mV)
exp = qa.variance_explained(trains, means)
self.assertAlmostEqual(exp[1][0], 1)
self.assertAlmostEqual(exp[1][0], 1)
self.assertAlmostEqual(exp[0][0], 1)
self.assertAlmostEqual(exp[0][1], 0.75)
def test_trains(self):
trains = {}
trains[0] = neo.SpikeTrain(
sp.zeros(2) * pq.ms,
0 * pq.ms,
waveforms=[[[-1, -2], [1, 2]], [[-1, -2], [1, 2]]] * pq.mV)
trains[1] = trains[0]
means = {}
means[1] = neo.Spike(0 * pq.ms, waveform=[[-1, -1], [1, 1]] * pq.mV)
exp = qa.variance_explained(trains, means)
self.assertAlmostEqual(exp[0][0], 1)
self.assertAlmostEqual(exp[0][0], 1)
self.assertAlmostEqual(exp[1][0], 1)
self.assertAlmostEqual(exp[1][1], 0.75)
