Commit f3c54d6f authored by Fred Drake's avatar Fred Drake

Add a test for the insertion of user-provided ADD_INFO records.

parent 165b2cc2
......@@ -41,8 +41,7 @@ class HotShotTestCase(unittest.TestCase):
return hotshot.Profile(self.logfn, lineevents, linetimings)
def get_logreader(self):
log = UnlinkingLogReader(self.logfn)
return log
return UnlinkingLogReader(self.logfn)
def get_events_wotime(self):
L = []
......@@ -68,6 +67,17 @@ class HotShotTestCase(unittest.TestCase):
profiler.close()
self.check_events(events)
def test_addinfo(self):
def f(p):
p.addinfo("test-key", "test-value")
profiler = self.new_profiler()
profiler.runcall(f, profiler)
profiler.close()
log = self.get_logreader()
info = log._info
list(log)
self.failUnless(info["test-key"] == ["test-value"])
def test_line_numbers(self):
def f():
y = 2
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment