1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
import numpy as np
from unittest.mock import patch
from numpy.testing import assert_allclose
import einsteinpy
def Timelike_test(metric, metric_params, position, momentum, steps, delta):
partial_lambdas = np.array([0, 1, 2, 3])
partial_vecs = np.array(
[
[
1.03068069e00,
3.99997742e01,
9.58510673e-02,
2.44928680e-15,
-9.79146613e-01,
-3.87363444e-04,
5.62571365e-19,
3.83405000e00,
],
[
2.06136190e00,
3.99987445e01,
1.91699898e-01,
2.44924485e-15,
-9.79146730e-01,
-9.48048366e-04,
1.12515772e-18,
3.83404999e00,
],
[
3.09204405e00,
3.99971561e01,
2.87545822e-01,
2.44918275e-15,
-9.79146663e-01,
-1.21478762e-03,
1.68776376e-18,
3.83405001e00,
],
[
4.12272819e00,
3.99949454e01,
3.83388599e-01,
2.44909661e-15,
-9.79146558e-01,
-1.80682765e-03,
2.25040630e-18,
3.83405000e00,
],
]
)
return partial_lambdas, partial_vecs
def test_precession_attr():
with patch("einsteinpy.examples.Timelike", new=Timelike_test):
p = einsteinpy.examples.precession()
L = p[1][0, -1]
assert p[0].shape[0] == p[1].shape[0]
assert p[1].shape[1] == 8
assert_allclose(p[1][:, 6], 0, atol=1e-12, rtol=1e-12)
assert_allclose(p[1][:, 7], L, atol=1e-6, rtol=1e-6)
|
