Fix in tests

xsuite · Nov 7, 2023 · f3afba5 · f3afba5
1 parent 709bdae
commit f3afba5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_tracker.py b/tests/test_tracker.py
@@ -535,6 +535,7 @@ def test_tracking_with_progress(test_context, with_progress, turns, collective):
     particles = xp.Particles(x=[1e-3, 2e-3, 3e-3], p0c=7e12, _context=test_context)
     line.build_tracker(_context=test_context)
     line.track(particles, num_turns=turns, with_progress=with_progress)
+    particles.move(xo.ContextCpu())
 
     assert np.all(particles.at_turn == turns)
     assert np.allclose(particles.s, 10 * turns, rtol=0, atol=1e-14)