Added assert

xsuite · Apr 8, 2021 · 7cd96f5 · 7cd96f5
1 parent bf7b42b
commit 7cd96f5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/examples/002_beambeam/000_beambeam.py b/examples/002_beambeam/000_beambeam.py
@@ -129,6 +129,11 @@
 
 bb_b1_pyst.track(p_pyst)
 
+assert np.allclose(p_pyst.px,
+    p2np(particles_b1.px[:n_probes]))
+assert np.allclose(p_pyst.py,
+    p2np(particles_b1.py[:n_probes]))
+
 import matplotlib.pyplot as plt
 plt.close('all')
 plt.figure()