File tree Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Original file line number Diff line number Diff line change 110110positions_x = positions_x [indices ]
111111positions_y = positions_y [indices ]
112112
113- assert np .allclose (z ,zetas )
113+ assert np .allclose (z / sigma_z ,zetas / sigma_z )
114114assert np .allclose (x / sigma_x ,positions_x )
115115assert np .allclose (y / sigma_y ,positions_y )
116116
117-
118117scaling_constant = particles .q0 ** 2 * cst .e ** 2 / (particles .p0c [0 ] * particles .beta0 [0 ] * cst .e )
119118for 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
You can’t perform that action at this time.
0 commit comments