def test_linear_map_cleans_slices(self):
'''Tests whether the slice_sets are deleted when the track() method
of the LinearMap is called
'''
circumference = 1.
Qs = 0.012
linear_map = LinearMap(self.alpha_array, circumference, Qs,
printer=AccumulatorPrinter())
beam = self.create_all1_bunch()
sliceset_mock = {'mock': 42}
beam._slice_sets = sliceset_mock
linear_map.track(beam)
self.assertFalse(beam._slice_sets, 'slice_sets not deleted after ' +
'calling LinearMap.track() [changing the ' +
'z-coordinates of the beam]')
def test_linear_map_track(self):
'''Tests whether only the dp and z coordinates are modified when
applying the track() method of LinearMap
'''
circumference = 1.
Qs = 0.012
linear_map = LinearMap(self.alpha_array, circumference, Qs)
beam = self.create_all1_bunch()
beam2 = self.create_all1_bunch()
linear_map.track(beam)
self.assertTrue(np.allclose(beam.x,beam2.x),
'x coord of beam has changed in LinearMap.track()')
self.assertTrue(np.allclose(beam.y,beam2.y),
'y coord of beam has changed in LinearMap.track()')
self.assertTrue(np.allclose(beam.xp,beam2.xp),
'xp coord of beam has changed in LinearMap.track()')
self.assertTrue(np.allclose(beam.yp,beam2.yp),
'yp coord of beam has changed in LinearMap.track()')
