test: use existing timestamp in test_parse_special_field

Calculating our the timestamp for the index of events is not needed.  Tell
trappy to not normalize the time and that way we don't need to do the
error-prone calculation ourselves.

1 files changed, 6 insertions, 7 deletions

diff --git a/tests/test_base.py b/tests/test_base.py
index 5fa3d78..5d3b716 100644
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -123,17 +123,16 @@ class TestBase(utils_tests.SetupDirectory):
             fout.write(in_data)
 
         ftrace_parser = trappy.register_dynamic_ftrace("Event0", "event0", scope="sched")
-        trace = trappy.FTrace()
+        trace = trappy.FTrace(normalize_time=False)
         dfr = trace.event0.data_frame
 
         self.assertEquals(set(dfr.columns), expected_columns)
 
-        for idx in range(len(events)):
-            timestamp = str( dfr.index[idx]+float(events.keys()[0]))
-            self.assertTrue(timestamp in events)
-            self.assertEquals(dfr["__comm"].iloc[idx], events[timestamp]['task'])
-            self.assertEquals(dfr["__pid"].iloc[idx],  events[timestamp]['pid'])
-            self.assertEquals(dfr["__cpu"].iloc[idx],  events[timestamp]['cpu'])
+        for timestamp, event in events.iteritems():
+            timestamp = float(timestamp)
+            self.assertEquals(dfr["__comm"].loc[timestamp], event['task'])
+            self.assertEquals(dfr["__pid"].loc[timestamp],  event['pid'])
+            self.assertEquals(dfr["__cpu"].loc[timestamp],  event['cpu'])
 
         trappy.unregister_dynamic_ftrace(ftrace_parser)