def test_neuron_connection():
SIMULATION_LENGTH = 5000
neuron_pre = brain.Neuron(baseline_rate=0.005)
neuron_post = brain.Neuron(baseline_rate=0.000)
connection = brain.Connection(amplitude=1, latency=5, rise_time=1, decay_time=1)
postsynaptic_input = np.zeros(SIMULATION_LENGTH)
for i in range(SIMULATION_LENGTH):
is_firing = neuron_pre.act(i)
if is_firing:
connection.act(i, postsynaptic_input)
neuron_post.act(i, probability_input=postsynaptic_input[i])
spk_train_pre = neuron_pre.get_action_history()
spk_train_post = neuron_post.get_action_history()
ccg, t = util.discreat_crosscorrelation(np.array(spk_train_pre), np.array(spk_train_post))
assert np.argmax(ccg) == 15
def test_discreat_crosscorrelation():
ts_trigger = np.arange(5)
ts_reference = np.arange(5) + 0.1001
ccg, t = util.discreat_crosscorrelation(ts_trigger, ts_reference, t_range=(-0.2, 0.5), bin_width=0.1)
assert(np.array_equal(ccg, [0, 0, 0, 5, 0, 0, 0]))
