diff options
| -rw-r--r-- | chroma/io/root.py | 1 | ||||
| -rw-r--r-- | test/test_io.py | 2 |
2 files changed, 2 insertions, 1 deletions
diff --git a/chroma/io/root.py b/chroma/io/root.py index b135df9..0454524 100644 --- a/chroma/io/root.py +++ b/chroma/io/root.py @@ -174,6 +174,7 @@ class RootWriter(object): if pyev.primary_vertex.pol is not None: self.ev.primary_vertex.pol.SetXYZ(*pyev.primary_vertex.pol) self.ev.primary_vertex.ke = pyev.primary_vertex.ke + self.ev.primary_vertex.t0 = pyev.primary_vertex.t0 if pyev.photons_beg is not None: photons = pyev.photons_beg diff --git a/test/test_io.py b/test/test_io.py index 3553058..15daab6 100644 --- a/test/test_io.py +++ b/test/test_io.py @@ -6,7 +6,7 @@ import numpy as np class TestRootIO(unittest.TestCase): def test_file_write_and_read(self): ev = event.Event(1, event.Vertex('e-', pos=(0,0,1), dir=(1,0,0), - ke=15.0, pol=(0,1,0))) + ke=15.0, pol=(0,1,0), t0=40.0)) photons_beg = root.make_photon_with_arrays(1) photons_beg.pos[0] = (1,2,3) |