def test_compare_kerr_and_schwarzschild_metric():
r = 99.9
theta = 5 * np.pi / 6
Rs = 9e-3
# Kerr metric would reduce to Schwarzschild metric under limits a=0
mk = kerr_utils.metric(_c, r, theta, Rs, 0.0)
ms = schwarzschild_utils.metric(_c, r, theta, Rs)
assert_allclose(mk, ms, rtol=1e-8)
def test_compare_kerr_and_schwarzschild_metric():
r = 99.9
theta = 5 * np.pi / 6
M = 6.73317655e26
# Kerr metric would reduce to Schwarzschild metric under limits a=0
mk = kerr_utils.metric(r, theta, M, 0.0)
ms = schwarzschild_utils.metric(r, theta, M)
assert_allclose(mk, ms, rtol=1e-8)
def test_calculate_trajectory(coords, time, M, start_lambda, end_lambda,
OdeMethodKwargs):
cl = Schwarzschild.from_spherical(coords, M, time)
ans = cl.calculate_trajectory(
start_lambda=start_lambda,
end_lambda=end_lambda,
OdeMethodKwargs=OdeMethodKwargs,
)
ans = ans[1]
testarray = list()
for i in ans:
g = schwarzschild_utils.metric(i[1], i[2], M.value)
testarray.append(g[0][0] * (i[4]**2) + g[1][1] * (i[5]**2) + g[2][2] *
(i[6]**2) + g[3][3] * (i[7]**2))
testarray = np.array(testarray, dtype=float)
comparearray = np.ones(shape=ans[:, 4].shape, dtype=float)
assert_allclose(testarray, comparearray, 1e-4)
def test_calculate_trajectory(pos_vec, vel_vec, time, M, start_lambda,
end_lambda, OdeMethodKwargs):
cl = Schwarzschild.from_spherical(pos_vec, vel_vec, time, M)
ans = cl.calculate_trajectory(
start_lambda=start_lambda,
end_lambda=end_lambda,
OdeMethodKwargs=OdeMethodKwargs,
)
_c, _scr = constant.c.value, schwarzschild_radius(M).value
ans = ans[1]
testarray = list()
for i in ans:
g = schwarzschild_utils.metric(_c, i[1], i[2], _scr)
testarray.append(g[0][0] * (i[4]**2) + g[1][1] * (i[5]**2) + g[2][2] *
(i[6]**2) + g[3][3] * (i[7]**2))
testarray = np.array(testarray, dtype=float)
comparearray = np.ones(shape=ans[:, 4].shape, dtype=float)
assert_allclose(testarray, comparearray, 1e-4)