Update test

next-exp · Jun 10, 2018 · a5f4215 · a5f4215
1 parent 32ae527
commit a5f4215
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/invisible_cities/database/test_data/Kr83_nexus_v5_03_00_ACTIVE_7bar_10evts_KDST.h5 b/invisible_cities/database/test_data/Kr83_nexus_v5_03_00_ACTIVE_7bar_10evts_KDST.h5
diff --git a/invisible_cities/io/kdst_io_test.py b/invisible_cities/io/kdst_io_test.py
@@ -29,6 +29,16 @@ def dump_df(write, df):
                         getattr(evt, col).append(value)
                     except AttributeError:
                         setattr(evt, col, value)
+
+            evt.nS1 = int(evt.nS1)
+            evt.nS2 = int(evt.nS2)
+            for col in ("Z", "DT"):
+                column = list(getattr(evt, col))
+                setattr(evt, col, [])
+                for i in range(evt.nS2):
+                    s1_data = column[:evt.nS1 ]
+                    column  = column[ evt.nS1:]
+                    getattr(evt, col).append(s1_data)
             write(evt)
 
     with tb.open_file(filename, 'w') as h5out: