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)