Skip to content

Commit c1a2174

Browse files
committed
fixing test for damper
1 parent 9334205 commit c1a2174

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

PyHEADTAIL/feedback/transverse_damper.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,13 +26,16 @@ def __init__(self, dampingrate_x, dampingrate_y, *args, **kwargs):
2626
self.prints('Damper in Y active')
2727
elif not dampingrate_x and not dampingrate_y:
2828
self.prints('Dampers not active')
29-
self.track = lambda x: 0
3029
else:
3130
self.gain_x = 2/dampingrate_x
3231
self.gain_y = 2/dampingrate_y
3332
self.track = self.track_all
3433
self.prints('Dampers active')
3534

35+
# will be overwritten at initialisation
36+
def track(self, beam):
37+
pass
38+
3639
def track_horizontal(self, beam):
3740
beam.xp -= self.gain_x * beam.mean_xp()
3841

PyHEADTAIL/testing/unittests/test_gpu_interface.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -296,7 +296,7 @@ def test_transverse_damper(self):
296296
bunch_gpu = self.create_all1_bunch()
297297
dampingrate_x = 0.01
298298
dampingrate_y = 0.05
299-
damp = TransverseDamper(dampingrate_x, dampingrate_y)
299+
damp = TransverseDamper(dampingrate_x, dampingrate_y, printer=SilentPrinter())
300300
self.assertTrue(self._track_cpu_gpu([damp], bunch_cpu, bunch_gpu),
301301
'Tracking TransverseDamper CPU/GPU differs')
302302

0 commit comments

Comments
 (0)