コード例 #1
0
def test_calc_pos_and_dir_along_movement():
    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)

    # check start and end at the right place
    assert_almost_equal(pos[...,0], start)
    assert_almost_equal(pos[...,-1], stop)
コード例 #2
0
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())