2222pytest .importorskip ("scipy" )
2323
2424
25- def fake_state (source , betas , ages ):
25+ def fake_state (source , betas ):
2626 """Fake dynamics: particles are displaced beta * 1e4 km from the source for
2727 a little more control on the tests."""
2828 r0 = 10000 * u .km
@@ -31,7 +31,7 @@ def fake_state(source, betas, ages):
3131 return State (
3232 (betas * r0 )[:, np .newaxis ] + source .r ,
3333 (betas * v0 )[:, np .newaxis ] + source .v ,
34- source .t - ages ,
34+ source .t ,
3535 frame = source .frame ,
3636 )
3737
@@ -55,8 +55,8 @@ def fake_syndynes():
5555
5656 items = []
5757 for beta in betas :
58- particles = fake_state (comet , np .repeat (beta , 10 ), ages * 0 )
59- initial = fake_state (comet , np .zeros (10 ), ages )
58+ particles = fake_state (comet , np .repeat (beta , 10 ))
59+ initial = fake_state (comet , np .zeros (10 ))
6060 items .append (
6161 Syndyne (
6262 comet ,
@@ -92,8 +92,8 @@ def fake_synchrones():
9292
9393 items = []
9494 for age in ages :
95- particles = fake_state (comet , betas , age )
96- initial = fake_state (comet , np .zeros (10 ), ages )
95+ particles = fake_state (comet , betas )
96+ initial = fake_state (comet , np .zeros (10 ))
9797 items .append (
9898 Synchrone (
9999 comet ,
@@ -499,7 +499,7 @@ def test_plot(self):
499499 assert all (line .get_ydata () <= 4.17e-3 * betas [i ])
500500
501501 # cannot plot relative to source when observer is None
502- state = fake_state (comet , [1 ], ages )
502+ state = fake_state (comet , [1 ])
503503 syndyne = Syndyne (
504504 comet , [1 ], ages , state .r , state .v , state .t , comet , observer = None
505505 )
@@ -556,7 +556,7 @@ def test_dt(self):
556556 comet = State ([1 , 0 , 0 ] * u .au , [0 , 30 , 0 ] * u .km / u .s , Time ("2023-12-07" ))
557557 dt = [- 1 , 0 , 1 ] * u .day
558558 ages = - dt
559- state = fake_state (comet , [0 , 1 , 2 ], ages )
559+ state = fake_state (comet , [0 , 1 , 2 ])
560560 orbit = SourceOrbit (comet , dt , state .r , state .v , comet .t + dt )
561561
562562 assert all (orbit .dt == dt )
0 commit comments