def test_BLtoCartesian_vel():
pos_vec = np.array([20, np.pi / 2, np.pi / 4])
vel_vec = np.array([0.0, 0.0, 10])
a = 0.0
ans_vec = np.array([-200 / np.sqrt(2), 200 / np.sqrt(2), 0])
ans = utils.BLToCartesian_vel(pos_vec, vel_vec, a)
assert_allclose(ans, ans_vec, rtol=0.0, atol=1e-5)
def test_BL2C_8dim(vec, a):
list1 = list()
for v in vec:
nv = np.hstack((
v[0],
utils.BLToCartesian_pos(v[1:4], a),
v[4],
utils.BLToCartesian_vel(v[1:4], v[5:8], a),
))
list1.append(nv)
arr1 = np.array(list1)
arr2 = utils.BL2C_8dim(vec, a)
assert_allclose(arr1, arr2, rtol=0.0, atol=1e-5)
def test_BL2C_8dim():
vec = np.array([
[21.0, 300, 0.33, 4.0, 1.0, -10.0, -1, -2],
[1.0, 3, 0.1, 5.0, 1.0, 100.0, -11, 2],
[1.0, 3, 0.1, 0, 1.0, 0, -11, 2],
])
a = 0.4
list1 = list()
for v in vec:
nv = np.hstack((
v[0],
utils.BLToCartesian_pos(v[1:4], a),
v[4],
utils.BLToCartesian_vel(v[1:4], v[5:8], a),
))
list1.append(nv)
arr1 = np.array(list1)
arr2 = utils.BL2C_8dim(vec, a)
assert_allclose(arr1, arr2, rtol=0.0, atol=1e-5)
def test_BLtoCartesian_vel(pos_vec, vel_vec, a, ans_vec):
ans = utils.BLToCartesian_vel(pos_vec, vel_vec, a)
assert_allclose(ans, ans_vec, rtol=0.0, atol=1e-5)
def test_cycle_vel(pos_vec, vel_vec, a):
pos_vec2 = utils.CartesianToBL_pos(pos_vec, a)
vel_vec2 = utils.CartesianToBL_vel(pos_vec, vel_vec, a)
# pos_vec3 = utils.BLToCartesian_pos(pos_vec2, a)
vel_vec3 = utils.BLToCartesian_vel(pos_vec2, vel_vec2, a)
assert_allclose(vel_vec, vel_vec3, rtol=0.0, atol=1e-5)