xtrack 0.89.1__tar.gz → 0.89.2__tar.gz

{xtrack-0.89.1/xtrack.egg-info → xtrack-0.89.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xtrack
-Version: 0.89.1
+Version: 0.89.2
 Summary: Tracking library for particle accelerators
 Home-page: https://xsuite.readthedocs.io/
 Download-URL: https://pypi.python.org/pypi/xtrack

{xtrack-0.89.1 → xtrack-0.89.2}/tests/test_tapering.py

@@ -29,7 +29,7 @@ def test_tapering_and_twiss_with_radiation():
 
     # Initial twiss (no radiation)
     line.configure_radiation(model=None)
-    tw_no_rad = line.twiss(method='4d', freeze_longitudinal=True)
+    tw_no_rad = line.twiss(method='4d')
 
     assert tw_no_rad.radiation_method == None
 
@@ -145,7 +145,7 @@ def test_tapering_zero_mean():
     line.particle_ref.p0c = 4e9  # eV
 
     line.configure_radiation(model=None)
-    tw_no_rad = line.twiss(method='4d', freeze_longitudinal=True)
+    tw_no_rad = line.twiss(method='4d')
 
     ###############################################
     # Enable radiation and compensate energy loss #

xtrack-0.89.2/xtrack/_version.py

@@ -0,0 +1 @@
+__version__ = '0.89.2'

{xtrack-0.89.1 → xtrack-0.89.2}/xtrack/base_element.py

@@ -15,6 +15,7 @@ from xobjects.hybrid_class import _build_xofields_dict
 from .general import _pkg_root
 from .internal_record import RecordIdentifier, RecordIndex, generate_get_record
 from .particles import Particles
+from .track_flags import c_header_flag_mapping
 
 start_per_part_block = """
     {
@@ -280,6 +281,7 @@ def _generate_per_particle_kernel_from_local_particle_function(
             for (int64_t batch_id = 0; batch_id < num_threads; batch_id++) {                   //only_for_context cpu_openmp
                 LocalParticle lpart;
                 lpart.io_buffer = io_buffer;
+                lpart.track_flags = 0;
                 int64_t part_id = batch_id * chunk_size;                                       //only_for_context cpu_openmp
                 int64_t end_id = (batch_id + 1) * chunk_size;                                  //only_for_context cpu_openmp
                 if (end_id > num_particles_to_track) end_id = num_particles_to_track;          //only_for_context cpu_openmp
@@ -452,6 +454,7 @@ class MetaBeamElement(xo.MetaHybridClass):
             _pkg_root.joinpath('headers','particle_states.h'),
             _pkg_root.joinpath('beam_elements', 'elements_src', 'track_srotation.h'),
             _pkg_root.joinpath('beam_elements', 'elements_src', 'track_drift.h'),
+            c_header_flag_mapping
         ]
         kernels = {}
 

{xtrack-0.89.1 → xtrack-0.89.2}/xtrack/beam_elements/elements_src/dipoleedge.h

@@ -18,22 +18,21 @@ void DipoleEdge_track_local_particle(DipoleEdgeData el, LocalParticle* part0){
 
     int64_t const model = DipoleEdgeData_get_model(el);
 
-    #ifdef XTRACK_MULTIPOLE_NO_SYNRAD
-        double const delta_taper = 0.0;
-    #else
-        #ifndef XTRACK_DIPOLEEDGE_TAPER
-        double const delta_taper = DipoleEdgeData_get_delta_taper(el);
-        #endif
+    double delta_taper = 0.0;
+    #ifndef XTRACK_MULTIPOLE_NO_SYNRAD
+        if (!LocalParticle_check_track_flag(part0, XS_FLAG_SR_TAPER)){
+            delta_taper = DipoleEdgeData_get_delta_taper(el);
+        }
     #endif
 
     if (model == 0){
         double r21 = DipoleEdgeData_get_r21(el);
         double r43 = DipoleEdgeData_get_r43(el);
 
-        #ifndef XTRACK_DIPOLEEDGE_TAPER
+        if (!LocalParticle_check_track_flag(part0, XS_FLAG_SR_TAPER)){
             r21 = r21 * (1 + delta_taper);
             r43 = r43 * (1 + delta_taper);
-        #endif
+        }
 
         #ifdef XSUITE_BACKTRACK
             r21 = -r21;
@@ -41,11 +40,11 @@ void DipoleEdge_track_local_particle(DipoleEdgeData el, LocalParticle* part0){
         #endif
 
         START_PER_PARTICLE_BLOCK(part0, part);
-            #ifdef XTRACK_DIPOLEEDGE_TAPER
+            if (LocalParticle_check_track_flag(part0, XS_FLAG_SR_TAPER)){
                 double const delta_taper = LocalParticle_get_delta(part);
                 r21 = r21 * (1 + delta_taper);
                 r43 = r43 * (1 + delta_taper);
-            #endif
+            }
 
             DipoleEdgeLinear_single_particle(part, r21, r43);
         END_PER_PARTICLE_BLOCK;

{xtrack-0.89.1 → xtrack-0.89.2}/xtrack/beam_elements/elements_src/track_magnet.h

@@ -400,12 +400,12 @@ void track_magnet_particles(
     #endif
 
     // Tapering
-    #ifdef XTRACK_MULTIPOLE_TAPER // Computing the tapering
+    if (LocalParticle_check_track_flag(part0, XS_FLAG_SR_TAPER)){
         part0->ipart = 0;
         delta_taper = LocalParticle_get_delta(part0); // I can use part0 because
                                                       // there is only one particle
                                                       // when doing the tapering
-    #endif
+    }
 
     #ifndef XTRACK_MULTIPOLE_NO_SYNRAD
         if (radiation_flag){

{xtrack-0.89.1 → xtrack-0.89.2}/xtrack/beam_elements/elements_src/track_magnet.template.h

@@ -319,12 +319,12 @@ void track_magnet_particles(
     #endif
 
     // Tapering
-    #ifdef XTRACK_MULTIPOLE_TAPER // Computing the tapering
+    if (LocalParticle_check_track_flag(part0, XS_FLAG_SR_TAPER)){
         part0->ipart = 0;
         delta_taper = LocalParticle_get_delta(part0); // I can use part0 because
                                                       // there is only one particle
                                                       // when doing the tapering
-    #endif
+    }
 
     #ifndef XTRACK_MULTIPOLE_NO_SYNRAD
         if (radiation_flag){

{xtrack-0.89.1 → xtrack-0.89.2}/xtrack/headers/synrad_spectrum.h

@@ -46,33 +46,33 @@ void synrad_average_kick(LocalParticle* part, double B_T, double lpath,
     }
     #endif
 
-    #ifdef XTRACK_SYNRAD_KICK_SAME_AS_FIRST
-    if (part -> ipart == 0){
-      *dp_record = LocalParticle_get_delta(part);
-      *dpx_record = LocalParticle_get_px(part);
-      *dpy_record = LocalParticle_get_py(part);
-    }
-    else{
-      f_t = 1.0;
+    if (LocalParticle_check_track_flag(part, XS_FLAG_SR_KICK_SAME_AS_FIRST)){
+      if (part -> ipart == 0){
+        *dp_record = LocalParticle_get_delta(part);
+        *dpx_record = LocalParticle_get_px(part);
+        *dpy_record = LocalParticle_get_py(part);
+      }
+      else{
+        f_t = 1.0;
+      }
     }
-    #endif
 
     LocalParticle_update_delta(part, (delta+1) * f_t - 1);
     LocalParticle_scale_px(part, f_t);
     LocalParticle_scale_py(part, f_t);
 
-    #ifdef XTRACK_SYNRAD_KICK_SAME_AS_FIRST
-    if (part -> ipart == 0){
-      *dp_record = LocalParticle_get_delta(part) - *dp_record;
-      *dpx_record = LocalParticle_get_px(part) - *dpx_record;
-      *dpy_record = LocalParticle_get_py(part) - *dpy_record;
-    }
+    if (LocalParticle_check_track_flag(part, XS_FLAG_SR_KICK_SAME_AS_FIRST)){
+      if (part -> ipart == 0){
+        *dp_record = LocalParticle_get_delta(part) - *dp_record;
+        *dpx_record = LocalParticle_get_px(part) - *dpx_record;
+        *dpy_record = LocalParticle_get_py(part) - *dpy_record;
+      }
     else{
       LocalParticle_update_delta(part, LocalParticle_get_delta(part) + *dp_record);
       LocalParticle_add_to_px(part, *dpx_record);
       LocalParticle_add_to_py(part, *dpy_record);
     }
-    #endif
+  }
 }
 
 GPUFUN

{xtrack-0.89.1 → xtrack-0.89.2}/xtrack/tapering.py

@@ -72,9 +72,8 @@ def compensate_radiation_energy_loss(line, delta0='zero_mean', rtol_eneloss=1e-1
     f_setter.set_values(np.zeros_like(f_setter.get_values()))
 
     if verbose: _print("Share energy loss among cavities (repeat until energy loss is zero)")
-    with xt.line._preserve_config(line):
-        line.config.XTRACK_MULTIPOLE_TAPER = True
-        line.config.XTRACK_DIPOLEEDGE_TAPER = True
+    with xt.line._preserve_track_flags(line):
+        line.tracker.track_flags.XS_FLAG_SR_TAPER = True
 
         if delta0 == 'zero_mean':
             num_rounds = 2

{xtrack-0.89.1 → xtrack-0.89.2}/xtrack/track_flags.py

@@ -2,18 +2,20 @@ import numpy as np
 
 flag_mapping = {
     'XS_FLAG_BACKTRACK': 0,
-    'XS_FLAG_TAPER': 1,
     'XS_FLAG_KILL_CAVITY_KICK': 2,
     'XS_FLAG_IGNORE_GLOBAL_APERTURE': 3,
-    'XS_FLAG_IGNORE_LOCAL_APERTURE': 4
+    'XS_FLAG_IGNORE_LOCAL_APERTURE': 4,
+    'XS_FLAG_SR_TAPER': 5,
+    'XS_FLAG_SR_KICK_SAME_AS_FIRST': 6
 }
 
 flag_defaults = {
     'XS_FLAG_BACKTRACK': False,
-    'XS_FLAG_TAPER': False,
     'XS_FLAG_KILL_CAVITY_KICK': False,
     'XS_FLAG_IGNORE_GLOBAL_APERTURE': False,
-    'XS_FLAG_IGNORE_LOCAL_APERTURE': False
+    'XS_FLAG_IGNORE_LOCAL_APERTURE': False,
+    'XS_FLAG_SR_TAPER': False,
+    'XS_FLAG_SR_KICK_SAME_AS_FIRST': False
 }
 
 c_header_flag_mapping = """

{xtrack-0.89.1 → xtrack-0.89.2}/xtrack/twiss.py

@@ -527,10 +527,9 @@ def twiss_line(line, particle_ref=None, method=None,
         assert radiation_method in ['full', 'kick_as_co', 'scale_as_co']
         assert freeze_longitudinal is False
         if (radiation_method == 'kick_as_co' and (
-            not hasattr(line.config, 'XTRACK_SYNRAD_KICK_SAME_AS_FIRST') or
-            not line.config.XTRACK_SYNRAD_KICK_SAME_AS_FIRST)):
-            with xt.line._preserve_config(line):
-                line.config.XTRACK_SYNRAD_KICK_SAME_AS_FIRST = True
+            not line.tracker.track_flags.XS_FLAG_SR_KICK_SAME_AS_FIRST)):
+            with xt.line._preserve_track_flags(line):
+                line.tracker.track_flags.XS_FLAG_SR_KICK_SAME_AS_FIRST = True
                 return _add_action_in_res(twiss_line(**kwargs), input_kwargs)
         elif (radiation_method == 'scale_as_co' and (
             not hasattr(line.config, 'XTRACK_SYNRAD_SCALE_SAME_AS_FIRST') or
@@ -540,8 +539,7 @@ def twiss_line(line, particle_ref=None, method=None,
                 return _add_action_in_res(twiss_line(**kwargs), input_kwargs)
 
     if radiation_method == 'kick_as_co':
-        assert hasattr(line.config, 'XTRACK_SYNRAD_KICK_SAME_AS_FIRST')
-        assert line.config.XTRACK_SYNRAD_KICK_SAME_AS_FIRST
+        assert line.tracker.track_flags.XS_FLAG_SR_KICK_SAME_AS_FIRST
 
     if line.enable_time_dependent_vars:
         raise RuntimeError('Time dependent variables not supported in Twiss')
@@ -731,7 +729,7 @@ def twiss_line(line, particle_ref=None, method=None,
             with xt.line._preserve_track_flags(line):
                 line.tracker.track_flags.XS_FLAG_KILL_CAVITY_KICK = True
                 line.config.XTRACK_MULTIPOLE_NO_SYNRAD = True
-                line.config.XTRACK_SYNRAD_KICK_SAME_AS_FIRST = False
+                line.tracker.track_flags.XS_FLAG_SR_KICK_SAME_AS_FIRST = False
                 cols_chrom, scalars_chrom = _compute_chromatic_functions(
                     line=line,
                     init=init,
@@ -778,7 +776,8 @@ def twiss_line(line, particle_ref=None, method=None,
             raise ValueError('method="4d" not supported for eneloss_and_damping=True')
         if radiation_method != 'full' or twiss_res._data['R_matrix_ebe'] is None:
             with xt.line._preserve_config(line):
-                line.config.XTRACK_SYNRAD_KICK_SAME_AS_FIRST = False
+              with xt.line._preserve_track_flags(line):
+                line.tracker.track_flags.XS_FLAG_SR_KICK_SAME_AS_FIRST = False
                 line.config.XTRACK_SYNRAD_SCALE_SAME_AS_FIRST = False
                 _, RR, _, _, _, RR_ebe = _find_periodic_solution(
                     line=line, particle_on_co=particle_on_co,
@@ -1380,7 +1379,7 @@ def _compute_coupling_elements_edwards_teng(
         CC = RR[2:4, :2]
         DD = RR[2:4, 2:4]
 
-        if np.linalg.norm(BB) < 1e-12 and np.linalg.norm(CC) < 1e-12:
+        if np.linalg.norm(BB) < 1e-10 and np.linalg.norm(CC) < 1e-10:
             R_edw_teng = np.zeros((2, 2))
         else:
             tr = np.linalg.trace
@@ -1602,7 +1601,7 @@ def _compute_chromatic_functions(line, init, delta_chrom, steps_r_matrix,
                 include_collective=include_collective)
             part_chrom = line.find_closed_orbit(delta0=dd, co_guess=part_guess,
                                     start=start, end=end, num_turns=num_turns,
-                                    symmetrize=(periodic_mode == 'periodic_symmetric'),
+                                    symmetrize=False,
                                     include_collective=include_collective,
                                     )
             tw_init_chrom.particle_on_co = part_chrom
@@ -1610,7 +1609,7 @@ def _compute_chromatic_functions(line, init, delta_chrom, steps_r_matrix,
                                         particle_on_co=tw_init_chrom.particle_on_co.copy(),
                                         start=start, end=end, num_turns=num_turns,
                                         steps_r_matrix=steps_r_matrix,
-                                        symmetrize=(periodic_mode == 'periodic_symmetric'),
+                                        symmetrize=False,
                                         include_collective=include_collective,
                                         )['R_matrix']
             (WW_chrom, _, _, _) = lnf.compute_linear_normal_form(RR_chrom,
@@ -2113,6 +2112,9 @@ def _find_periodic_solution(line, particle_on_co, particle_ref, method,
 
     assert periodic_mode in ['periodic', 'periodic_symmetric']
 
+    if periodic_mode == 'periodic_symmetric':
+        raise ValueError('`periodic_symmetric` not supported anymore')
+
     if start is not None or end is not None:
         assert start is not None and end is not None, (
             'start and end must be both None or both not None')
@@ -2124,6 +2126,7 @@ def _find_periodic_solution(line, particle_on_co, particle_ref, method,
         delta0 = 0
 
     if periodic_mode == 'periodic_symmetric':
+        raise ValueError('`periodic_symmetric` not supported anymore')
         assert R_matrix is None, 'R_matrix must be None for `periodic_symmetric`'
         assert W_matrix is None, 'W_matrix must be None for `periodic_symmetric`'
 
@@ -2146,7 +2149,7 @@ def _find_periodic_solution(line, particle_on_co, particle_ref, method,
                                 search_for_t_rev=search_for_t_rev,
                                 spin=spin,
                                 num_turns_search_t_rev=num_turns_search_t_rev,
-                                symmetrize=(periodic_mode == 'periodic_symmetric'),
+                                symmetrize=False,
                                 include_collective=include_collective
                                 )
     if only_orbit:
@@ -2177,7 +2180,7 @@ def _find_periodic_solution(line, particle_on_co, particle_ref, method,
                     num_turns=num_turns,
                     element_by_element=compute_R_element_by_element,
                     only_markers=only_markers,
-                    symmetrize=(periodic_mode == 'periodic_symmetric'),
+                    symmetrize=False,
                     include_collective=include_collective
                     )
                 RR = RR_out['R_matrix']
@@ -2703,9 +2706,7 @@ def compute_one_turn_matrix_finite_differences(
         assert end is not None
         line.track(part_temp, ele_start=start, ele_stop=end)
         if symmetrize:
-            with xt.line._preserve_config(line):
-                line.config.XSUITE_MIRROR = True
-                line.track(part_temp, ele_start=start, ele_stop=end)
+            raise NotImplementedError
     elif particle_on_co._xobject.at_element[0]>0:
         assert element_by_element is False, 'Not yet implemented'
         assert num_turns == 1, 'Not yet implemented'
@@ -2721,9 +2722,7 @@ def compute_one_turn_matrix_finite_differences(
         line.track(part_temp, num_turns=num_turns,
                    turn_by_turn_monitor=monitor_setting)
         if symmetrize:
-            with xt.line._preserve_config(line):
-                line.config.XSUITE_MIRROR = True
-                line.track(part_temp, num_turns=num_turns)
+            raise NotImplementedError
 
     temp_mat = np.zeros(shape=(6, 13), dtype=np.float64)
     temp_mat[0, :] = context.nparray_from_context_array(part_temp.x)

{xtrack-0.89.1 → xtrack-0.89.2/xtrack.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xtrack
-Version: 0.89.1
+Version: 0.89.2
 Summary: Tracking library for particle accelerators
 Home-page: https://xsuite.readthedocs.io/
 Download-URL: https://pypi.python.org/pypi/xtrack

{xtrack-0.89.1 → xtrack-0.89.2}/xtrack.egg-info/SOURCES.txt

@@ -77,7 +77,6 @@ tests/test_optimize_for_tracking.py
 tests/test_particles.py
 tests/test_particles_basics.py
 tests/test_particles_pdg.py
-tests/test_periodic_symmetric_twiss_and_match.py
 tests/test_pipeline.py
 tests/test_prebuild_kernels.py
 tests/test_ps_against_ptc.py

xtrack-0.89.1/tests/test_periodic_symmetric_twiss_and_match.py

@@ -1,163 +0,0 @@
-import xtrack as xt
-import xobjects as xo
-import numpy as np
-
-def test_periodic_symmetric_twiss_and_match():
-
-    # Build line with half a cell
-    half_cell = xt.Line(
-        elements={
-            'start_cell': xt.Marker(),
-            'drift0': xt.Drift(length=1.),
-            'qf1': xt.Quadrupole(k1=0.027/2, length=1.),
-            'drift1_1': xt.Drift(length=1),
-            'bend1': xt.Bend(k0=3e-4, h=3e-4, length=45.),
-            'drift1_2': xt.Drift(length=1.),
-            'qd1': xt.Quadrupole(k1=-0.0271/2, length=1.),
-            'drift2': xt.Drift(length=1),
-            'mid_cell': xt.Marker(),
-        }
-    )
-    half_cell.particle_ref = xt.Particles(p0c=2e9)
-
-    tw_half_cell = half_cell.twiss4d(init='periodic_symmetric', # <--- periodic-symmetric boundary
-                                    strengths=True # to get the strengths in table
-                                    )
-
-    cell = xt.Line(
-        elements={
-            'start_cell': xt.Marker(),
-            'drift0': xt.Drift(length=1.),
-            'qf1':    xt.Quadrupole(k1=0.027/2, length=1.),
-            'drift1': xt.Drift(length=1),
-            'bend1':  xt.Bend(k0=3e-4, h=3e-4, length=45.),
-            'drift2': xt.Drift(length=1.),
-            'qd1':    xt.Quadrupole(k1=-0.0271/2, length=1.),
-            'drift3': xt.Drift(length=1),
-            'mid_cell': xt.Marker(),
-            'drift4': xt.Replica('drift3'),
-            'qd2':    xt.Replica('qd1'),
-            'drift5': xt.Replica('drift2'),
-            'bend2':  xt.Replica('bend1'),
-            'drift6': xt.Replica('drift1'),
-            'qf2':    xt.Replica('qf1'),
-            'drift7': xt.Replica('drift0'),
-            'end_cell': xt.Marker(),
-        }
-    )
-    cell.particle_ref = xt.Particles(p0c=2e9)
-    tw_cell = cell.twiss4d(strengths=True)
-
-    xo.assert_allclose(tw_half_cell.betx[:-1], # remove '_end_point'
-                    tw_cell.rows[:'mid_cell'].betx, atol=0, rtol=1e-8)
-    xo.assert_allclose(tw_half_cell.bety[:-1], # remove '_end_point'
-                    tw_cell.rows[:'mid_cell'].bety, atol=0, rtol=1e-8)
-    xo.assert_allclose(tw_half_cell.alfx[:-1], # remove '_end_point'
-                    tw_cell.rows[:'mid_cell'].alfx, atol=1e-8, rtol=0)
-    xo.assert_allclose(tw_half_cell.alfy[:-1], # remove '_end_point'
-                    tw_cell.rows[:'mid_cell'].alfy, atol=1e-8, rtol=0)
-    xo.assert_allclose(tw_half_cell.dx[:-1], # remove '_end_point'
-                    tw_cell.rows[:'mid_cell'].dx, atol=1e-8, rtol=0)
-    xo.assert_allclose(tw_half_cell.dpx[:-1], # remove '_end_point'
-                        tw_cell.rows[:'mid_cell'].dpx, atol=1e-8, rtol=0)
-
-    xo.assert_allclose(tw_half_cell.ax_chrom[:-1], # remove '_end_point'
-                        tw_cell.rows[:'mid_cell'].ax_chrom, atol=1e-5, rtol=0)
-    xo.assert_allclose(tw_half_cell.ay_chrom[:-1], # remove '_end_point'
-                        tw_cell.rows[:'mid_cell'].ay_chrom, atol=1e-5, rtol=0)
-    xo.assert_allclose(tw_half_cell.bx_chrom[:-1], # remove '_end_point'
-                        tw_cell.rows[:'mid_cell'].bx_chrom, atol=1e-5, rtol=0)
-    xo.assert_allclose(tw_half_cell.by_chrom[:-1], # remove '_end_point'
-                        tw_cell.rows[:'mid_cell'].by_chrom, atol=1e-5, rtol=0)
-
-    xo.assert_allclose(tw_half_cell.qx, tw_cell.qx / 2, atol=1e-9, rtol=0)
-    xo.assert_allclose(tw_half_cell.qy, tw_cell.qy / 2, atol=1e-9, rtol=0)
-    xo.assert_allclose(tw_half_cell.dqx, tw_cell.dqx / 2, atol=1e-6, rtol=0)
-    xo.assert_allclose(tw_half_cell.dqy, tw_cell.dqy / 2, atol=1e-6, rtol=0)
-
-    for ll in [cell, half_cell]:
-        ll.vars['kqf'] = 0.027/2
-        ll.vars['kqd'] = -0.0271/2
-        ll.element_refs['qf1'].k1 = ll.vars['kqf']
-        ll.element_refs['qd1'].k1 = ll.vars['kqd']
-
-    tw_cell = cell.twiss4d(strengths=True)
-    tw_half_cell = half_cell.twiss4d(init='periodic_symmetric', strengths=True)
-
-    xo.assert_allclose(tw_cell.mux[-1], 0.231, atol=1e-3, rtol=0)
-    xo.assert_allclose(tw_cell.muy[-1], 0.233, atol=1e-3, rtol=0)
-    xo.assert_allclose(tw_half_cell.mux[-1], 0.231 / 2, atol=1e-3, rtol=0)
-    xo.assert_allclose(tw_half_cell.muy[-1], 0.233 / 2, atol=1e-3, rtol=0)
-
-    opt_halfcell = half_cell.match(
-        method='4d',
-        start='start_cell', end='mid_cell',
-        init='periodic_symmetric',
-        targets=xt.TargetSet(mux=0.2501/2, muy=0.2502/2, at='mid_cell'),
-        vary=xt.VaryList(['kqf', 'kqd'], step=1e-5),
-    )
-
-    tw_cell = cell.twiss4d(strengths=True)
-    tw_half_cell = half_cell.twiss4d(init='periodic_symmetric', strengths=True)
-
-    xo.assert_allclose(tw_half_cell.mux[-1], 0.2501 / 2, atol=1e-3, rtol=0)
-    xo.assert_allclose(tw_half_cell.muy[-1], 0.2502 / 2, atol=1e-3, rtol=0)
-    xo.assert_allclose(tw_cell.mux[-1], 0.231, atol=1e-3, rtol=0) # unaffected
-    xo.assert_allclose(tw_cell.muy[-1], 0.233, atol=1e-3, rtol=0) # unaffected
-
-    opt_cell = cell.match(
-        method='4d',
-        start='start_cell', end='end_cell',
-        init='periodic',
-        targets=xt.TargetSet(mux=0.2501, muy=0.2502, at='end_cell'),
-        vary=xt.VaryList(['kqf', 'kqd'], step=1e-5),
-    )
-
-    xo.assert_allclose(half_cell.vv['kqf'], cell.vv['kqf'], rtol=1e-9, atol=0)
-    xo.assert_allclose(half_cell.vv['kqd'], cell.vv['kqd'], rtol=1e-9, atol=0)
-
-    tw_cell = cell.twiss4d(strengths=True)
-    tw_half_cell = half_cell.twiss4d(init='periodic_symmetric', strengths=True)
-
-    xo.assert_allclose(tw_half_cell.mux[-1], 0.2501 / 2, atol=1e-3, rtol=0)
-    xo.assert_allclose(tw_half_cell.muy[-1], 0.2502 / 2, atol=1e-3, rtol=0)
-    xo.assert_allclose(tw_cell.mux[-1], 0.2501, atol=1e-3, rtol=0)
-    xo.assert_allclose(tw_cell.muy[-1], 0.2502, atol=1e-3, rtol=0)
-
-    xo.assert_allclose(tw_half_cell.betx[:-1], # remove '_end_point'
-                    tw_cell.rows[:'mid_cell'].betx, atol=0, rtol=1e-8)
-    xo.assert_allclose(tw_half_cell.bety[:-1], # remove '_end_point'
-                    tw_cell.rows[:'mid_cell'].bety, atol=0, rtol=1e-8)
-    xo.assert_allclose(tw_half_cell.alfx[:-1], # remove '_end_point'
-                    tw_cell.rows[:'mid_cell'].alfx, atol=1e-8, rtol=0)
-    xo.assert_allclose(tw_half_cell.alfy[:-1], # remove '_end_point'
-                    tw_cell.rows[:'mid_cell'].alfy, atol=1e-8, rtol=0)
-    xo.assert_allclose(tw_half_cell.dx[:-1], # remove '_end_point'
-                    tw_cell.rows[:'mid_cell'].dx, atol=1e-8, rtol=0)
-    xo.assert_allclose(tw_half_cell.dpx[:-1], # remove '_end_point'
-                        tw_cell.rows[:'mid_cell'].dpx, atol=1e-8, rtol=0)
-    xo.assert_allclose(tw_half_cell.ddx[:-1], # remove '_end_point'
-                   tw_cell.rows[:'mid_cell'].ddx, atol=1e-7, rtol=1e-8)
-
-    xo.assert_allclose(tw_half_cell.ax_chrom[:-1], # remove '_end_point'
-                        tw_cell.rows[:'mid_cell'].ax_chrom, atol=1e-5, rtol=0)
-    xo.assert_allclose(tw_half_cell.ay_chrom[:-1], # remove '_end_point'
-                        tw_cell.rows[:'mid_cell'].ay_chrom, atol=1e-5, rtol=0)
-    xo.assert_allclose(tw_half_cell.bx_chrom[:-1], # remove '_end_point'
-                        tw_cell.rows[:'mid_cell'].bx_chrom, atol=1e-5, rtol=0)
-    xo.assert_allclose(tw_half_cell.by_chrom[:-1], # remove '_end_point'
-                        tw_cell.rows[:'mid_cell'].by_chrom, atol=1e-5, rtol=0)
-    xo.assert_allclose(tw_half_cell.ddx[:-1], # remove '_end_point'
-                        tw_cell.rows[:'mid_cell'].ddx, atol=1e-7, rtol=1e-8)
-
-    xo.assert_allclose(tw_half_cell.qx, tw_cell.qx / 2, atol=1e-9, rtol=0)
-    xo.assert_allclose(tw_half_cell.qy, tw_cell.qy / 2, atol=1e-9, rtol=0)
-    xo.assert_allclose(tw_half_cell.dqx, tw_cell.dqx / 2, atol=1e-6, rtol=0)
-    xo.assert_allclose(tw_half_cell.dqy, tw_cell.dqy / 2, atol=1e-6, rtol=0)
-
-    tw_off_mom_cell = cell.twiss4d(strengths=True, delta0=1e-3)
-    tw_off_mom_half_cell = half_cell.twiss4d(
-        init='periodic_symmetric', strengths=True, delta0=1e-3)
-
-    xo.assert_allclose(tw_off_mom_half_cell.x[:-1],
-                    tw_off_mom_cell.rows[:'mid_cell'].x, atol=1e-12, rtol=0)

xtrack-0.89.1/xtrack/_version.py

@@ -1 +0,0 @@
-__version__ = '0.89.1'