class TestTrackIn(TestCase):
def setUp(self):
self.track_in = TrackIn((1, 2, 3, 4, 0, 0, 1, 8, 9, 10, 11),
zed_correction=0)
def test_trackin_init(self):
track_in = self.track_in
self.assertEqual(1, track_in.id)
self.assertListEqual([2, 3, 4], list(track_in.pos))
self.assertListEqual([0, 0, 1], list(track_in.dir))
self.assertEqual(8, track_in.E)
self.assertEqual(9, track_in.time)
self.assertEqual(130, track_in.particle_type) # this should be PDG!
self.assertEqual(11, track_in.length)
def test_track_repr(self):
repr_str = ("Track:\n id: 1\n pos: [2 3 4]\n dir: (0.0, 0.0, 1.0)\n"
" energy: 8 GeV\n time: 9 ns\n type: 130 'K0L' [PDG]\n"
" length: 11 [m]\n")
self.assertEqual(repr_str, self.track_in.__repr__())
def setUp(self):
self.track_in = TrackIn((1, 2, 3, 4, 0, 0, 1, 8, 9, 10, 11),
zed_correction=0)
