def test_generate_cell_properties(): PD, PG = fm.generate_cell_properties() assert_almost_equal(np.sum(PD**2), 1.) PDk = [1., 0.3, 0.2] PDe = PDk / np.linalg.norm(PDk) assert np.allclose(PDe, PD) assert_almost_equal(np.sum(PG**2), 1.)
def test_generate_firing_rate(): time, disp, speed, acc = fm.generate_ballistic_profiles() #start = np.array((0., 0., 0.)) stop = np.array((1., 1., 1.)) pos, drn = fm.calc_pos_and_dir_along_movement(disp, stop) PD, PG = fm.generate_cell_properties() k = fm.generate_firing_rate(pos, drn, speed, acc, time, \ PD, PG, model='k') assert_(np.all(k == 1.)) ds = fm.generate_firing_rate(pos, drn, speed, acc, time, \ PD, PG, model='ds') kds = fm.generate_firing_rate(pos, drn, speed, acc, time, \ PD, PG, model='kds') assert_(np.allclose(ds + k, kds)) assert_(kds.size == pos.shape[1] - 1) assert_(kds.argmax() == speed.argmax())