def test_binarize_no_time(self):
st = self.test_array_1d
times = np.arange(0, 1.23 + .01, .01)
target = np.zeros_like(times).astype('bool')
for time in self.test_array_1d:
target[get_nearest(times, time)] = True
res0, tres = cv.binarize(st, return_times=True, sampling_rate=100)
res1 = cv.binarize(st, return_times=False, sampling_rate=100)
res2 = cv.binarize(st, sampling_rate=100)
assert_array_almost_equal(res0, res1, decimal=9)
assert_array_almost_equal(res0, res2, decimal=9)
def test_binarize_no_time(self):
st = self.test_array_1d
times = np.arange(0, 1.23 + .01, .01)
target = np.zeros_like(times).astype('bool')
for time in self.test_array_1d:
target[get_nearest(times, time)] = True
res0, tres = cv.binarize(st, return_times=True, sampling_rate=100)
res1 = cv.binarize(st, return_times=False, sampling_rate=100)
res2 = cv.binarize(st, sampling_rate=100)
assert_array_almost_equal(res0, res1, decimal=9)
assert_array_almost_equal(res0, res2, decimal=9)
def test_binarize_with_quantities_round_set_ends(self):
st = pq.Quantity(self.test_array_1d, units='ms')
times = np.arange(5., 10. + .1, .1)
target = np.zeros_like(times).astype('bool')
times = pq.Quantity(times, units='ms')
res, tres = cv.binarize(st, return_times=True, t_start=5., t_stop=10.,
sampling_rate=10.*pq.kHz)
assert_array_almost_equal(res, target, decimal=9)
assert_array_almost_equal(tres, times, decimal=9)
def test_binarize_with_spiketrain_exact_set_ends(self):
st = neo.SpikeTrain(self.test_array_1d, units='ms',
t_stop=10.0, sampling_rate=100)
times = np.arange(5., 10. + .01, .01)
target = np.zeros_like(times).astype('bool')
times = pq.Quantity(times, units='ms')
res, tres = cv.binarize(st, return_times=True, t_start=5., t_stop=10.)
assert_array_almost_equal(res, target, decimal=9)
assert_array_almost_equal(tres, times, decimal=9)
def test_binarize_with_plain_array_exact_set_ends(self):
st = self.test_array_1d
times = np.arange(0, 10. + .01, .01)
target = np.zeros_like(times).astype('bool')
for time in self.test_array_1d:
target[get_nearest(times, time)] = True
res, tres = cv.binarize(st, return_times=True, t_stop=10., sampling_rate=100.)
assert_array_almost_equal(res, target, decimal=9)
assert_array_almost_equal(tres, times, decimal=9)
def test_binarize_with_plain_array_exact(self):
st = self.test_array_1d
times = np.arange(0, 1.23 + .01, .01)
target = np.zeros_like(times).astype('bool')
for time in self.test_array_1d:
target[get_nearest(times, time)] = True
res, tres = cv.binarize(st, return_times=True, sampling_rate=100)
assert_array_almost_equal(res, target, decimal=9)
assert_array_almost_equal(tres, times, decimal=9)
def test_binarize_with_quantities_round_set_ends(self):
st = pq.Quantity(self.test_array_1d, units='ms')
times = np.arange(5., 10. + .1, .1)
target = np.zeros_like(times).astype('bool')
times = pq.Quantity(times, units='ms')
res, tres = cv.binarize(st, return_times=True, t_start=5., t_stop=10.,
sampling_rate=10. * pq.kHz)
assert_array_almost_equal(res, target, decimal=9)
assert_array_almost_equal(tres, times, decimal=9)
def test_binarize_with_spiketrain_exact_set_ends(self):
st = neo.SpikeTrain(self.test_array_1d, units='ms',
t_stop=10.0, sampling_rate=100)
times = np.arange(5., 10. + .01, .01)
target = np.zeros_like(times).astype('bool')
times = pq.Quantity(times, units='ms')
res, tres = cv.binarize(st, return_times=True, t_start=5., t_stop=10.)
assert_array_almost_equal(res, target, decimal=9)
assert_array_almost_equal(tres, times, decimal=9)
def test_binarize_with_quantities_exact_set_ends(self):
st = pq.Quantity(self.test_array_1d, units='ms')
times = np.arange(0, 10. + .01, .01)
target = np.zeros_like(times).astype('bool')
for time in self.test_array_1d:
target[get_nearest(times, time)] = True
times = pq.Quantity(times, units='ms')
res, tres = cv.binarize(st, return_times=True, t_stop=10.,
sampling_rate=100.*pq.kHz)
assert_array_almost_equal(res, target, decimal=9)
assert_array_almost_equal(tres, times, decimal=9)
def test_binarize_with_spiketrain_round(self):
st = neo.SpikeTrain(self.test_array_1d, units='ms',
t_stop=10.0, sampling_rate=10.0)
times = np.arange(0, 10. + .1, .1)
target = np.zeros_like(times).astype('bool')
for time in np.round(self.test_array_1d, 1):
target[get_nearest(times, time)] = True
times = pq.Quantity(times, units='ms')
res, tres = cv.binarize(st, return_times=True)
assert_array_almost_equal(res, target, decimal=9)
assert_array_almost_equal(tres, times, decimal=9)
def test_binarize_with_quantities_exact_set_ends(self):
st = pq.Quantity(self.test_array_1d, units='ms')
times = np.arange(0, 10. + .01, .01)
target = np.zeros_like(times).astype('bool')
for time in self.test_array_1d:
target[get_nearest(times, time)] = True
times = pq.Quantity(times, units='ms')
res, tres = cv.binarize(st, return_times=True, t_stop=10.,
sampling_rate=100. * pq.kHz)
assert_array_almost_equal(res, target, decimal=9)
assert_array_almost_equal(tres, times, decimal=9)
def test_binarize_with_spiketrain_round(self):
st = neo.SpikeTrain(self.test_array_1d, units='ms',
t_stop=10.0, sampling_rate=10.0)
times = np.arange(0, 10. + .1, .1)
target = np.zeros_like(times).astype('bool')
for time in np.round(self.test_array_1d, 1):
target[get_nearest(times, time)] = True
times = pq.Quantity(times, units='ms')
res, tres = cv.binarize(st, return_times=True)
assert_array_almost_equal(res, target, decimal=9)
assert_array_almost_equal(tres, times, decimal=9)
def conversion_binarize(spiketrain, sampling_rate=None, t_start=None,
t_stop=None, return_times=None):
result = binarize(spiketrain, sampling_rate, t_start, t_stop,
return_times)
# print result
return result
