Skip to content

Commit 6e2eaf7

Browse files
committed
Fix issue with too relaxxed absolute tolerance in test
1 parent b38ef00 commit 6e2eaf7

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

tests/test_fakemultibunch.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -110,17 +110,16 @@
110110
positions_x = positions_x[indices]
111111
positions_y = positions_y[indices]
112112

113-
assert np.allclose(z,zetas)
113+
assert np.allclose(z/sigma_z,zetas/sigma_z)
114114
assert np.allclose(x/sigma_x,positions_x)
115115
assert np.allclose(y/sigma_y,positions_y)
116116

117-
118117
scaling_constant = particles.q0**2 * cst.e**2 / (particles.p0c[0] * particles.beta0[0] * cst.e)
119118
for i_slice in range(num_slices):
120119
zetas_slice = zeta0[i_slice]-zetas
121120
kicks = positions_x*sigma_x*scaling_constant*slice_intensity*wfx.function_vs_zeta(zetas_slice,beta0=betar,dzeta=moments_data.dz)
122-
assert np.isclose(np.sum(kicks),kicks_x_from_track[i_slice])
123-
kicks = positions_x*sigma_x*scaling_constant*slice_intensity*wfy.function_vs_zeta(zetas_slice,beta0=betar,dzeta=moments_data.dz)
124-
assert np.isclose(np.sum(kicks),kicks_y_from_track[i_slice])
121+
assert np.isclose(np.sum(kicks),kicks_x_from_track[i_slice],atol=1E-15)
122+
kicks = positions_y*sigma_y*scaling_constant*slice_intensity*wfy.function_vs_zeta(zetas_slice,beta0=betar,dzeta=moments_data.dz)
123+
assert np.isclose(np.sum(kicks),kicks_y_from_track[i_slice],atol=1E-15)
125124

126125

0 commit comments

Comments
 (0)