Skip to content

Commit 66eb424

Browse files
Fix random seeds in test_build_particles tests
1 parent 50347f6 commit 66eb424

File tree

3 files changed

+33
-27
lines changed

3 files changed

+33
-27
lines changed

tests/test_build_particles_binomial.py

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,11 @@
1313

1414
from xpart.longitudinal import generate_binomial_longitudinal_coordinates
1515

16-
from xobjects.test_helpers import for_all_test_contexts
16+
from xobjects.test_helpers import fix_random_seed, for_all_test_contexts
1717

1818

1919
@for_all_test_contexts
20+
@fix_random_seed(42343298)
2021
def test_build_particles_binomial(test_context):
2122
# Build a reference particle
2223
p0 = xp.Particles(mass0=xp.PROTON_MASS_EV, q0=1, p0c=7e12, x=1, y=3,
@@ -33,14 +34,15 @@ def test_build_particles_binomial(test_context):
3334
line.build_tracker(_context=test_context)
3435

3536
# Built a set of three particles with different x coordinates
36-
zeta, delta, matcher = generate_binomial_longitudinal_coordinates(num_particles=num_part,
37-
nemitt_x=3e-6,
38-
nemitt_y=3e-6,
39-
sigma_z=0.05,
40-
particle_ref=p0,
41-
line=line,
42-
return_matcher=True
43-
)
37+
zeta, delta, matcher = generate_binomial_longitudinal_coordinates(
38+
num_particles=num_part,
39+
nemitt_x=3e-6,
40+
nemitt_y=3e-6,
41+
sigma_z=0.05,
42+
particle_ref=p0,
43+
line=line,
44+
return_matcher=True,
45+
)
4446

4547
# Test if longitudinal coordinates match with Single
4648
# Generate distribution from RF matcher

tests/test_build_particles_parabolic.py

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,11 @@
1313

1414
from xpart.longitudinal import generate_parabolic_longitudinal_coordinates
1515

16-
from xobjects.test_helpers import for_all_test_contexts
16+
from xobjects.test_helpers import fix_random_seed, for_all_test_contexts
1717

1818

1919
@for_all_test_contexts
20+
@fix_random_seed(78292384)
2021
def test_build_particles_parabolic(test_context):
2122
# Build a reference particle
2223
p0 = xp.Particles(mass0=xp.PROTON_MASS_EV, q0=1, p0c=7e12, x=1, y=3,
@@ -33,14 +34,15 @@ def test_build_particles_parabolic(test_context):
3334
line.build_tracker(_context=test_context)
3435

3536
# Built a set of three particles with different x coordinates
36-
zeta, delta, matcher = generate_parabolic_longitudinal_coordinates(num_particles=num_part,
37-
nemitt_x=3e-6,
38-
nemitt_y=3e-6,
39-
sigma_z=0.05,
40-
particle_ref=p0,
41-
line=line,
42-
return_matcher=True
43-
)
37+
zeta, delta, matcher = generate_parabolic_longitudinal_coordinates(
38+
num_particles=num_part,
39+
nemitt_x=3e-6,
40+
nemitt_y=3e-6,
41+
sigma_z=0.05,
42+
particle_ref=p0,
43+
line=line,
44+
return_matcher=True,
45+
)
4446

4547
# Test if longitudinal coordinates match with Single
4648
# Generate distribution from RF matcher

tests/test_build_particles_qgaussian.py

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,11 @@
1313

1414
from xpart.longitudinal import generate_qgaussian_longitudinal_coordinates
1515

16-
from xobjects.test_helpers import for_all_test_contexts
16+
from xobjects.test_helpers import fix_random_seed, for_all_test_contexts
1717

1818

1919
@for_all_test_contexts
20+
@fix_random_seed(4237638)
2021
def test_build_particles_qgaussian(test_context):
2122
# Build a reference particle
2223
p0 = xp.Particles(mass0=xp.PROTON_MASS_EV, q0=1, p0c=7e12, x=1, y=3,
@@ -33,14 +34,15 @@ def test_build_particles_qgaussian(test_context):
3334
line.build_tracker(_context=test_context)
3435

3536
# Built a set of three particles with different x coordinates
36-
zeta, delta, matcher = generate_qgaussian_longitudinal_coordinates(num_particles=num_part,
37-
nemitt_x=3e-6,
38-
nemitt_y=3e-6,
39-
sigma_z=0.05,
40-
particle_ref=p0,
41-
line=line,
42-
return_matcher=True
43-
)
37+
zeta, delta, matcher = generate_qgaussian_longitudinal_coordinates(
38+
num_particles=num_part,
39+
nemitt_x=3e-6,
40+
nemitt_y=3e-6,
41+
sigma_z=0.05,
42+
particle_ref=p0,
43+
line=line,
44+
return_matcher=True,
45+
)
4446

4547
# Test if longitudinal coordinates match with Single
4648
# Generate distribution from RF matcher

0 commit comments

Comments
 (0)