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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
# Compare the C++ and Python implementations of the KerrBL metric
#
# This script is for demonstration and test purposes, in all practical case
# the C++ implementation should be preferred.
# Import gyoto, require the Python plug-in. Note that the Python
# plug-in name depends on the running interpreter. Default to basename
# of Python executable, then python3, then python.
import gyoto.core, gyoto.std
import sys, os.path, importlib, gyoto_sample_metrics
try:
gyoto.core.requirePlugin(os.path.basename(sys.executable))
except gyoto.core.Error:
try:
gyoto.core.requirePlugin("python3")
except gyoto.core.Error:
try:
gyoto.core.requirePlugin("python")
except gyoto.core.Error:
raise gyoto.core.Error("Could not load Python plugin, tried: "+os.path.basename(sys.executable)+", python3 and python")
# Instanciate an instance of the C++ KerrBL and an instance of the Python equivalent
kerrc=gyoto.std.KerrBL()
kerrp=gyoto.core.Metric("Python")
kerrp.set("Module", "gyoto_sample_metrics")
kerrp.set("Class", "KerrBL")
# Set spin
kerrc.spin(0.5)
kerrp.set("Parameters", (0.5,))
# Test getRms
Rmsp=kerrp.getRms()
Rmsc=kerrc.getRms()
assert (Rmsp==Rmsc), "Rms is different"
# Test getRmb
Rmbp=kerrp.getRmb()
Rmbc=kerrc.getRmb()
assert (Rmbp==Rmbc), "Rmb is different"
# Test getSpecificAngularMomentum
samp=kerrp.getSpecificAngularMomentum(10.)
samc=kerrc.getSpecificAngularMomentum(10.)
assert (samp==samc), "Specific angular momentum is different"
# Test gmunu
pos=[0, 10, 1, 1.]
ggp=kerrp.gmunu(pos)
ggc=kerrc.gmunu(pos)
assert (ggp==ggc).all(), "gmunu is different"
# Test christoffel
pos=[0, 10, 1, 1.]
chp=kerrp.christoffel(pos)
chc=kerrp.christoffel(pos)
assert (chp==chc).all(), "christoffel is different"
# Test getPotential
Wp=kerrp.getPotential([0, 10, 1, 1.], 10.)
Wc=kerrc.getPotential([0, 10, 1, 1.], 10.)
assert (Wp==Wc), "getPotential is different"
# Test isStopCondition
Wp=kerrp.isStopCondition([0, 10, 1, 1., 0., 0., 0., 0])
Wc=kerrc.isStopCondition([0, 10, 1, 1., 0., 0., 0., 0])
assert (Wp==Wc), "isStopCondition is different"
# Test circularVelocity
kerrp.keplerian(False)
kerrc.keplerian(False)
vp=kerrp.circularVelocity([0, 10, 1, 1.])
vc=kerrc.circularVelocity([0, 10, 1, 1.])
assert (vp==vc).all(), "isStopCondition is different"
vp=numpy.ndarray(4, dtype=float)
vc=numpy.ndarray(4, dtype=float)
kerrp.circularVelocity([0, 10, 1, 1.], vp)
kerrc.circularVelocity([0, 10, 1, 1.], vc)
assert (vp==vc).all(), "isStopCondition is different"
vp=numpy.ndarray(4, dtype=float)
vc=numpy.ndarray(4, dtype=float)
kerrp.circularVelocity([0, 10, 1, 1.], vp, -1)
kerrc.circularVelocity([0, 10, 1, 1.], vc, -1)
assert (vp==vc).all(), "isStopCondition is different"
print ("All tests passed successfully")
|
