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