diff options
Diffstat (limited to 'utils/timeline_test.py')
-rw-r--r-- | utils/timeline_test.py | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/utils/timeline_test.py b/utils/timeline_test.py new file mode 100644 index 00000000..1f4d178a --- /dev/null +++ b/utils/timeline_test.py @@ -0,0 +1,54 @@ +# Copyright 2012 Google Inc. All Rights Reserved. + +"""Tests for time_line.py.""" + +__author__ = 'yunlian@google.com (Yunlian Jiang)' + +import time +import unittest + +import timeline + + +class TimeLineTest(unittest.TestCase): + + def testRecord(self): + tl = timeline.Timeline() + tl.Record('A') + t = time.time() + t1 = tl.events[0].timestamp + self.assertEqual(int(t1-t), 0) + self.assertRaises(AssertionError, tl.Record, 'A') + + def testGetEvents(self): + tl = timeline.Timeline() + tl.Record('A') + e = tl.GetEvents() + self.assertEqual(e, ['A']) + tl.Record('B') + e = tl.GetEvents() + self.assertEqual(e, ['A', 'B']) + + def testGetEventTime(self): + tl = timeline.Timeline() + tl.Record('A') + t = time.time() + t1 = tl.GetEventTime('A') + self.assertEqual(int(t1-t), 0) + self.assertRaises(IndexError, tl.GetEventTime, 'B') + + def testGetLastEventTime(self): + tl = timeline.Timeline() + self.assertRaises(IndexError, tl.GetLastEventTime) + tl.Record('A') + t = time.time() + t1 = tl.GetLastEventTime() + self.assertEqual(int(t1-t), 0) + time.sleep(2) + tl.Record('B') + t = time.time() + t1 = tl.GetLastEventTime() + self.assertEqual(int(t1-t), 0) + +if __name__ == '__main__': + unittest.main() |