@@ -15,27 +15,27 @@ def distance_to(one_body: Body, other_body: Body) -> numpy.ndarray:
1515 >>> earth = Body.from_tuple_parameters("Earth", 5.972*(10**24), 100, (1.506*(10**11), 0, 100), (0, 29290, 0))
1616 >>> moon = Body.from_tuple_parameters("Moon", 0.0735*(10**24), 100, (1.496*(10**11), 384.4*(10**6), -500), (1050, 29290, 0))
1717 >>> distance_to(moon, earth)
18- array([[ 1.00000000e +09],
19- [ -3.84400000e +08],
20- [ 6.00000000e +02]])
18+ array([[ 1.000e +09],
19+ [-3.844e +08],
20+ [ 6.000e +02]])
2121
2222 Distance to itself is always 0 in all directions.
2323
2424 >>> moon = Body.from_tuple_parameters("Moon", 0.0735*(10**24), 100, (1.496*(10**11), 384.4*(10**6), -500), (1050, 29290, 0))
2525 >>> distance_to(moon, moon)
26- array([[ 0.],
27- [ 0.],
28- [ 0.]])
26+ array([[0.],
27+ [0.],
28+ [0.]])
2929
3030 >>> from nbp.helpers.numpy import tuple_to_numpy
3131 >>> import numpy as np
3232 >>> velocity = tuple_to_numpy((0, 0, 0))
3333 >>> one = Body('saturn', 100, 100, np.array([[0.], [0.], [0.]]), velocity)
3434 >>> two = Body('neptune', 100, 100, tuple_to_numpy((0, 146.2, 0)), velocity)
3535 >>> distance_to(one, two)
36- array([[ 0. ],
37- [ 146.2],
38- [ 0. ]])
36+ array([[ 0. ],
37+ [146.2],
38+ [ 0. ]])
3939 """
4040 return other_body .position - one_body .position
4141
@@ -49,12 +49,12 @@ def absolute_distance_to_one(one_body: Body, other_body: Body) -> float:
4949 >>> earth = Body.from_tuple_parameters("Earth", 5.972*(10**24), 100.0, (1.496*(10**11), 0, 0), (0, 29290, 0))
5050 >>> moon = Body.from_tuple_parameters("Moon", 0.0735*(10**24), 100.0, (1.496*(10**11), 384.4*(10**6), 0), (1050, 29290, 0))
5151 >>> absolute_distance_to_one(moon, earth)
52- 384400000.0
52+ np.float64( 384400000.0)
5353
5454 >>> earth = Body.from_tuple_parameters("Earth", 5.972*(10**24), 100.0, (1.496*(10**11), 0, 0), (0, 29290, 0))
5555 >>> moon = Body.from_tuple_parameters("Moon", 0.0735*(10**24), 100.0, (1.496*(10**11), -384.4*(10**6), -69834), (1050, 29290, 0))
5656 >>> absolute_distance_to_one(moon, earth)
57- 384400006.34337604
57+ np.float64( 384400006.34337604)
5858
5959 >>> from nbp.helpers.numpy import tuple_to_numpy
6060 >>> import numpy as np
@@ -107,9 +107,9 @@ def acceleration_to_all(one_body: Body, bodies: [Body]) -> numpy.ndarray:
107107 >>> moon = Body.from_tuple_parameters("Moon", 0.0735*(10**24), 100.0, (0, 384.4*(10**6), -1000), (0, 0, 0))
108108
109109 >>> acceleration_to_all(moon, [kg, earth1, earth2, moon])
110- array([[ -4.46878801e-05],
111- [ -5.48536334e-03],
112- [ 6.07257588e-08]])
110+ array([[-4.46878801e-05],
111+ [-5.48536334e-03],
112+ [ 6.07257588e-08]])
113113
114114 """
115115 total_acceleration = numpy .array ([[0 ],
@@ -130,9 +130,9 @@ def calculate_position(body: Body, delta_time: float) -> numpy.ndarray:
130130
131131 >>> test_body = Body.from_tuple_parameters("Test_body", 1.0, 1.0, (60, -20, 15), (4, 10.2, -6))
132132 >>> calculate_position(test_body, 3.0)
133- array([[ 72. ],
134- [ 10.6],
135- [ -3. ]])
133+ array([[72. ],
134+ [10.6],
135+ [-3. ]])
136136 """
137137 return body .position + (delta_time * body .velocity )
138138
@@ -151,17 +151,17 @@ def calculate_velocity(one_body: Body, delta_time: float, other_bodies: [Body])
151151 >>> e5 = Body.from_tuple_parameters("Earth5", (5.972*(10**24)), 100.0, (6371000, 9000, -532), (0, 0, 0))
152152 >>> e6 = Body.from_tuple_parameters("Earth6", (5.972*(10**24)), 100.0, (-6371000, -9000, 532), (0, 0, 0))
153153 >>> calculate_velocity(kg, 314.0, [kg, e1, e2, e3, e4, e5, e6])
154- array([[ 0.],
155- [ 0.],
156- [ 0.]])
154+ array([[0.],
155+ [0.],
156+ [0.]])
157157
158158 >>> kg = Body.from_tuple_parameters("kg", 1.0, 100.0, (0, 0, 0), (0, 0, 0))
159159 >>> earth1 = Body.from_tuple_parameters("Earth1", (5.972*(10**24)), 100.0, (0, 6371000, 6280), (0, 0, 0))
160160 >>> moon = Body.from_tuple_parameters("Moon", 0.0735*(10**24), 100.0, (0, 384.4*(10**6), -1000), (0, 0, 0))
161161 >>> calculate_velocity(kg, 16.0, [kg, earth1, moon])
162- array([[ 0.00000000e+00],
163- [ 1.57114698e+02],
164- [ 1.54870030e-01]])
162+ array([[0.00000000e+00],
163+ [1.57114698e+02],
164+ [1.54870030e-01]])
165165 """
166166
167167 return one_body .velocity + (delta_time * acceleration_to_all (one_body , other_bodies ))
@@ -179,14 +179,14 @@ def merge_bodies(one_body: Body, other_body: Body) -> Body:
179179 >>> planet = merge_bodies(earth, moon)
180180
181181 >>> planet.position
182- array([[ 1.50587842e+11],
183- [ 4.67395352e+06],
184- [ 9.27053180e+01]])
182+ array([[1.50587842e+11],
183+ [4.67395352e+06],
184+ [9.27053180e+01]])
185185
186186 >>> planet.velocity
187- array([[ -8.60185262e+01],
188- [ 2.85777959e+04],
189- [ -7.59904061e-01]])
187+ array([[-8.60185262e+01],
188+ [ 2.85777959e+04],
189+ [-7.59904061e-01]])
190190
191191 >>> planet.radius
192192 6413824.949215559
@@ -235,7 +235,7 @@ def minimal_distance(bodies: [Body]) -> float:
235235 >>> saturn = Body.from_tuple_parameters('saturn', 568000000000000000000000000, 100, (0, 1352550000000, 0), (0, 10180, 0))
236236 >>> neptune = Body.from_tuple_parameters('neptune', 102413000000000000000000000, 100, (0, -4444450000000, 500000), (-5370, 0, 0))
237237 >>> minimal_distance([sun, earth, moon, jupiter, saturn, neptune])
238- 405500037.33168757
238+ np.float64( 405500037.33168757)
239239
240240 >>> sun = Body.from_tuple_parameters('Sun', 1989000000000000000000000000000, 100, (0, 0, 0), (0, 0, 0))
241241 >>> earth = Body.from_tuple_parameters('earth', 5972000000000000000000000, 100, (0, 152100000000, 1000), (29290, 0, 32))
@@ -244,15 +244,15 @@ def minimal_distance(bodies: [Body]) -> float:
244244 >>> saturn = Body.from_tuple_parameters('saturn', 568000000000000000000000000, 100, (0, 1352550000000, 0), (0, 10180, 0))
245245 >>> neptune = Body.from_tuple_parameters('neptune', 102413000000000000000000000, 100, (0, -4444450000000, 500000), (-5370, 0, 0))
246246 >>> minimal_distance([sun, earth, moon, jupiter, saturn, neptune])
247- 405500037.76202041
247+ np.float64( 405500037.7620204)
248248
249249 >>> sun = Body.from_tuple_parameters('Sun', 1989000000000000000000000000000, 100, (0, 0, 0), (0, 0, 0))
250250 >>> earth = Body.from_tuple_parameters('earth', 5972000000000000000000000, 100, (107550941418, 107550941418, 100000), (29290, 0, 32))
251251 >>> jupiter = Body.from_tuple_parameters('jupiter', 1900000000000000000000000000, 100, (816620000000, 0, -1000), (40, 12440, 1))
252252 >>> saturn = Body.from_tuple_parameters('saturn', 568000000000000000000000000, 100, (0, 1352550000000, 0), (0, 10180, 0))
253253 >>> neptune = Body.from_tuple_parameters('neptune', 102413000000000000000000000, 100, (0, -4444450000000, 500000), (-5370, 0, 0))
254254 >>> minimal_distance([sun, earth, jupiter, saturn, neptune])
255- 152099999999.3627
255+ np.float64( 152099999999.3627)
256256
257257 >>> from nbp.helpers.numpy import tuple_to_numpy
258258 >>> import numpy as np
0 commit comments