Esempio n. 1
0
 def test_px(self):
     pt, eta, phi = convert_px_py_pz(1, 0, 0)
     self.assertTrue(pt == 1)
     self.assertTrue(phi == 0)
     self.assertTrue(eta == 0)
Esempio n. 2
0
 def test_pz_neg(self):
     pt, eta, phi = convert_px_py_pz(0, 0, -1)
     self.assertEqual(pt, 0)
     self.assertEqual(phi, 0)
     self.assertEqual(eta, -float('inf'))
Esempio n. 3
0
 def test_pxpy(self):
     pt, eta, phi = convert_px_py_pz(1, 1, 0)
     self.assertEqual(pt, math.sqrt(2))
     self.assertAlmostEqual(phi, math.pi / 4)
     self.assertEqual(eta, 0)
Esempio n. 4
0
 def test_py_neg(self):
     pt, eta, phi = convert_px_py_pz(0, -1, 0)
     self.assertEqual(pt, 1)
     self.assertAlmostEqual(phi, -math.pi / 2)
     self.assertEqual(eta, 0)