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)