def test_scale_vel_500():
res = scale_velocity(500, 0.2)
assert res == 0.1
def test_scale_vel_list():
res = scale_velocity([1, 2, 3, 4])
np.testing.assert_array_almost_equal(res,
np.array([0.005, 0.01, 0.015, 0.02]))
def test_scale_vel_1000():
res = scale_velocity(1000, 0.2)
assert res == 0.2
def test_scale_vel_2_len3():
res = scale_velocity(2, 3)
assert res == 0.006
def test_scale_vel_1_len3():
res = scale_velocity(1, 3)
assert res == 0.003
def test_scale_vel_1():
res = scale_velocity(1, 0.2)
assert res == 0.0002