def test_refperiod_violations(self):
t1 = sp.array([0, 5, 10, 12, 17, 18])
t2 = sp.array([20000, 18000, 14000, 9000, 0, 5000])
t3 = sp.array([1,2,3])
st1 = neo.SpikeTrain(t1*pq.s,20*pq.s)
st2 = neo.SpikeTrain(t2*pq.ms,20*pq.s)
st3 = neo.SpikeTrain(t3*pq.h,3*pq.h)
r1,r2 = qa.get_refperiod_violations({0: [st1], 1: [st2,st3]}, 3.0*pq.s)
self.assertEqual(r1, 3, 'Total number of refractory period violations incorrect')
assert_arrays_equal(r2[0][0], neo.SpikeTrain(sp.array([10,17])*pq.s,20*pq.s))
assert_arrays_equal(r2[1][0], neo.SpikeTrain(sp.array([18])*pq.s,20*pq.s))
assert_arrays_equal(r2[1][1], neo.SpikeTrain(sp.array([])*pq.s,10800*pq.ms))
def test_refperiod_violations(self):
t1 = sp.array([0, 5, 10, 12, 17, 18])
t2 = sp.array([20000, 18000, 14000, 9000, 0, 5000])
t3 = sp.array([1, 2, 3])
st1 = neo.SpikeTrain(t1 * pq.s, 20 * pq.s)
st2 = neo.SpikeTrain(t2 * pq.ms, 20 * pq.s)
st3 = neo.SpikeTrain(t3 * pq.h, 3 * pq.h)
r1, r2 = qa.get_refperiod_violations({
0: [st1],
1: [st2, st3]
}, 3.0 * pq.s)
self.assertEqual(
r1, 3, 'Total number of refractory period violations incorrect')
assert_arrays_equal(
r2[0][0], neo.SpikeTrain(sp.array([10, 17]) * pq.s, 20 * pq.s))
assert_arrays_equal(r2[1][0],
neo.SpikeTrain(sp.array([18]) * pq.s, 20 * pq.s))
assert_arrays_equal(r2[1][1],
neo.SpikeTrain(sp.array([]) * pq.s, 10800 * pq.ms))
def test_refperiod_violations_empty(self):
num, d = qa.get_refperiod_violations({}, 1*pq.ms)
self.assertEqual(num, 0)
self.assertEqual(d, {})
def test_refperiod_violations_empty(self):
num, d = qa.get_refperiod_violations({}, 1 * pq.ms)
self.assertEqual(num, 0)
self.assertEqual(d, {})
