def test_Ekin(self):
print('\nTest momentum')
beam = tlwall.Beam(p_MeV_c=0.1)
self.assertEqual(0.87, round(beam.betarel, 2))
self.assertEqual(2.00, round(beam.gammarel, 3))
self.assertEqual(938.3, round(beam.Ekin_MeV, 1))
self.assertEqual(0.1, round(beam.p_MeV_c, 1))
def test_default(self):
print('\nTesting default values')
beam = tlwall.Beam()
self.assertEqual(1, beam.betarel)
self.assertEqual(float('inf'), beam.gammarel)
self.assertEqual(float('inf'), beam.Ekin_MeV)
self.assertEqual(float('inf'), beam.p_MeV_c)
def test_Ekin(self):
print('\nTest kinetic energy')
beam = tlwall.Beam(Ekin_MeV=0.1)
self.assertEqual(0.015, round(beam.betarel, 3))
self.assertEqual(1.0, round(beam.gammarel, 3))
self.assertEqual(0.1, round(beam.Ekin_MeV, 1))
self.assertEqual(13.7, round(beam.p_MeV_c, 1))
def test_betarel(self):
print('\nTest beta')
beam = tlwall.Beam(betarel=0.106)
self.assertEqual(0.106, beam.betarel)
self.assertEqual(1.006, round(beam.gammarel, 3))
self.assertEqual(5.3, round(beam.Ekin_MeV, 1))
self.assertEqual(100.0, round(beam.p_MeV_c, 1))
def test_gammarel(self):
print('\nTest gamma')
beam = tlwall.Beam(gammarel=1.006)
self.assertEqual(0.109, round(beam.betarel, 3))
self.assertEqual(1.006, round(beam.gammarel, 3))
self.assertEqual(5.6, round(beam.Ekin_MeV, 1))
self.assertEqual(102.9, round(beam.p_MeV_c, 1))
def read_beam(self, cfg_file=None):
if cfg_file is not None:
config = self.read_cfg(cfg_file)
else:
config = self.config
test_beam_shift = config.getfloat('beam_info', 'test_beam_shift')
if config.has_option('beam_info', 'betarel'):
betarel = config.getfloat('beam_info', 'betarel')
if config.has_option('beam_info', 'mass_MeV_c2'):
mass_MeV_c2 = config.getfloat('beam_info', 'mass_MeV_c2')
beam = tlwall.Beam(betarel=betarel,
test_beam_shift=test_beam_shift,
mass_MeV_c2=mass_MeV_c2)
else:
beam = tlwall.Beam(betarel=betarel,
test_beam_shift=test_beam_shift)
return beam
if config.has_option('beam_info', 'gammarel'):
gammarel = config.getfloat('beam_info', 'gammarel')
if config.has_option('beam_info', 'mass_MeV_c2'):
mass_MeV_c2 = config.getfloat('beam_info', 'mass_MeV_c2')
beam = tlwall.Beam(gammarel=gammarel,
test_beam_shift=test_beam_shift,
mass_MeV_c2=mass_MeV_c2)
else:
beam = tlwall.Beam(gammarel=gammarel,
test_beam_shift=test_beam_shift)
return beam
if config.has_option('beam_info', 'Ekin_MeV'):
Ekin_MeV = config.getfloat('beam_info', 'Ekin_MeV')
if config.has_option('beam_info', 'mass_MeV_c2'):
mass_MeV_c2 = config.getfloat('beam_info', 'mass_MeV_c2')
beam = tlwall.Beam(Ekin_MeV=Ekin_MeV,
test_beam_shift=test_beam_shift,
mass_MeV_c2=mass_MeV_c2)
else:
beam = tlwall.Beam(Ekin_MeV=Ekin_MeV,
test_beam_shift=test_beam_shift)
return beam
if config.has_option('beam_info', 'p_MeV_c'):
p_MeV_c = config.getfloat('beam_info', 'p_MeV_c')
if config.has_option('beam_info', 'mass_MeV_c2'):
mass_MeV_c2 = config.getfloat('beam_info', 'mass_MeV_c2')
beam = tlwall.Beam(p_MeV_c=p_MeV_c,
test_beam_shift=test_beam_shift,
mass_MeV_c2=mass_MeV_c2)
else:
beam = tlwall.Beam(p_MeV_c=p_MeV_c,
test_beam_shift=test_beam_shift)
return beam
def test_momentum_neg(self):
print('\nTest negative value for momentum')
beam = tlwall.Beam(p_MeV_c=-4.99)
self.assertEqual(1, beam.betarel)
def test_Ekin_neg(self):
print('\nTest negative value for kinetic energy')
beam = tlwall.Beam(Ekin_MeV=-0.99)
self.assertEqual(1, beam.betarel)
def test_betarel_gt1(self):
print('\nTest beta greater than 1')
beam = tlwall.Beam(betarel=1.99)
self.assertEqual(1, beam.betarel)
def test_betarel_neg(self):
print('\nTest negative value for beta')
beam = tlwall.Beam(betarel=-0.99)
self.assertEqual(1, beam.betarel)
def test_string(self):
print('\nTest string value for beta')
beam = tlwall.Beam(betarel='betarel')
self.assertEqual(1, beam.betarel)