Skip to content

Commit 963510c

Browse files
committed
Merge branch 'gianni_release' into release/v0.6.1
2 parents d920fc5 + a696619 commit 963510c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_jaw_position.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,6 @@
1212
import xcoll as xc
1313
import time
1414

15-
import matplotlib.pyplot as plt
16-
1715
# TODO: particle angles
1816

1917
jaws = [0.001, [0.0013, -0.002789], [-1.2e-6, -3.2e-3], [3.789e-3, 4.678e-7]]
@@ -123,6 +121,8 @@ def _plot_jaws(coll, part_init, part, hit_ids, not_hit_ids):
123121
wrong_ids = part.parent_particle_id[mask]
124122
mask_wrong_init = np.isin(part_init.particle_id, wrong_ids)
125123

124+
import matplotlib.pyplot as plt
125+
126126
mask_hit_init = np.isin(part_init.particle_id, hit_ids)
127127
mask_not_hit_init = np.isin(part_init.particle_id, not_hit_ids)
128128
plt.scatter(part_init.x[mask_not_hit_init], part_init.y[mask_not_hit_init], c='b', s=2)

0 commit comments

Comments
 (0)