@@ -70,15 +70,15 @@ def test_create_initial_distribution(beam, npart,impact_parameter, pencil_spread
7070
7171 with flaky_assertions ():
7272 # Pencil: left jaw
73- pos_jawL_conv = coll_conv .jaw_L
73+ pos_jawL_conv = coll_conv .jaw_LU
7474 pos_partL_conv = part_conv .x [mask_conv_L ].min ()
7575 pencil_spread_convL = part_conv .x [mask_conv_L ].max () - pos_partL_conv
7676 assert np .isclose (pencil_spread_convL , pencil_spread , atol = atol_spread )
7777 assert pos_partL_conv - impact_parameter - pos_jawL_conv < atol_cut
7878 assert pos_partL_conv - impact_parameter - pos_jawL_conv > 0
7979
8080 # Pencil: right jaw
81- pos_jawR_conv = coll_conv .jaw_R
81+ pos_jawR_conv = coll_conv .jaw_RU
8282 pos_partR_conv = part_conv .x [mask_conv_R ].max ()
8383 pencil_spread_convR = pos_partR_conv - part_conv .x [mask_conv_R ].min ()
8484 assert np .isclose (pencil_spread_convR , pencil_spread , atol = atol_spread )
@@ -117,15 +117,15 @@ def test_create_initial_distribution(beam, npart,impact_parameter, pencil_spread
117117
118118 with flaky_assertions ():
119119 # Pencil: left jaw
120- pos_jawL_div = coll_div .jaw_L
120+ pos_jawL_div = coll_div .jaw_LD
121121 pos_partL_div = part_div .y [mask_div_L ].min ()
122122 pencil_spread_divL = part_div .y [mask_div_L ].max () - pos_partL_div
123123 assert np .isclose (pencil_spread_divL , pencil_spread , atol = atol_spread )
124124 assert pos_partL_div - impact_parameter - pos_jawL_div < atol_cut
125125 assert pos_partL_div - impact_parameter - pos_jawL_div > 0
126126
127127 # Pencil: right jaw
128- pos_jawR_div = coll_div .jaw_R
128+ pos_jawR_div = coll_div .jaw_RD
129129 pos_partR_div = part_div .y [mask_div_R ].max ()
130130 pencil_spread_divR = pos_partR_div - part_div .y [mask_div_R ].min ()
131131 assert np .isclose (pencil_spread_divR , pencil_spread , atol = atol_spread )
0 commit comments