def test_p4_pt_calc(self):
a = ost.FourMomentum((100, 0.0, 1, 1), 'x,y,z,e')
self.assertTrue(a.pt, 100.0)
a = ost.FourMomentum((0.0, 100.0, 1, 1), 'x,y,z,e')
self.assertTrue(a.pt, 100.0)
a = ost.FourMomentum((5.0, 5.0, 1, 1), 'x,y,z,e')
self.assertTrue(a.pt, 25.0)
def test_p4_addition(self):
a = ost.FourMomentum((100, 1, 1, 1), 'x,y,z,e')
b = ost.FourMomentum((50, 1, 1, 1), 'x,y,z,e')
c = a + b
self.assertTrue(c.px, a.px + b.px)
self.assertTrue(c.py, a.py + b.py)
self.assertTrue(c.pz, a.pz + b.pz)
self.assertTrue(c.e, a.e + b.e)
def test_el_mass(self):
el = ost.Particle(pdg_id=11,
momentum=ost.FourMomentum((1., 1., 1., 1.)))
self.assertEqual(el.mass(), 0.5e-3)
def test_p4_eta_calc(self):
a = ost.FourMomentum((100, 1, 1, 1), 'x,y,z,e')
eta = .5 * np.log(
(np.sqrt(100.0**2 + 2.0) + 1.) / (np.sqrt(100.0**2 + 2.0) - 1.))
self.assertTrue(a.eta, eta)
def test_mu_mass(self):
mu = ost.Particle(pdg_id=13,
momentum=ost.FourMomentum((1., 1., 1., 1.)))
self.assertEqual(mu.mass(), 106e-3)