mirror of
https://github.com/python/cpython.git
synced 2025-09-19 23:20:25 +00:00
c687960496
Remove the log file after we are done with it. This should clean up after the test even on Windows, since the file is now closed before we attempt removal.
94 lines
2.8 KiB
Python
94 lines
2.8 KiB
Python
import hotshot
|
|
import hotshot.log
|
|
import os
|
|
import pprint
|
|
import unittest
|
|
|
|
import test_support
|
|
|
|
from hotshot.log import ENTER, EXIT, LINE
|
|
|
|
|
|
def shortfilename(fn):
|
|
# We use a really shortened filename since an exact match is made,
|
|
# and the source may be either a Python source file or a
|
|
# pre-compiled bytecode file.
|
|
if fn:
|
|
return os.path.splitext(os.path.basename(fn))[0]
|
|
else:
|
|
return fn
|
|
|
|
|
|
class UnlinkingLogReader(hotshot.log.LogReader):
|
|
"""Extend the LogReader so the log file is unlinked when we're
|
|
done with it."""
|
|
|
|
def __init__(self, logfn):
|
|
self.__logfn = logfn
|
|
hotshot.log.LogReader.__init__(self, logfn)
|
|
|
|
def next(self, index=None):
|
|
try:
|
|
return hotshot.log.LogReader.next(self)
|
|
except (IndexError, StopIteration):
|
|
os.unlink(self.__logfn)
|
|
raise
|
|
|
|
|
|
class HotShotTestCase(unittest.TestCase):
|
|
def new_profiler(self, lineevents=0, linetimings=1):
|
|
self.logfn = test_support.TESTFN
|
|
return hotshot.Profile(self.logfn, lineevents, linetimings)
|
|
|
|
def get_logreader(self):
|
|
log = UnlinkingLogReader(self.logfn)
|
|
return log
|
|
|
|
def get_events_wotime(self):
|
|
L = []
|
|
for event in self.get_logreader():
|
|
what, (filename, lineno, funcname), tdelta = event
|
|
L.append((what, (shortfilename(filename), lineno, funcname)))
|
|
return L
|
|
|
|
def check_events(self, expected):
|
|
events = self.get_events_wotime()
|
|
if events != expected:
|
|
self.fail(
|
|
"events did not match expectation; got:\n%s\nexpected:\n%s"
|
|
% (pprint.pformat(events), pprint.pformat(expected)))
|
|
|
|
def run_test(self, callable, events, profiler=None):
|
|
if profiler is None:
|
|
profiler = self.new_profiler()
|
|
profiler.runcall(callable)
|
|
profiler.close()
|
|
self.check_events(events)
|
|
|
|
def test_line_numbers(self):
|
|
def f():
|
|
y = 2
|
|
x = 1
|
|
def g():
|
|
f()
|
|
f_lineno = f.func_code.co_firstlineno
|
|
g_lineno = g.func_code.co_firstlineno
|
|
events = [(ENTER, ("test_hotshot", g_lineno, "g")),
|
|
(LINE, ("test_hotshot", g_lineno, "g")),
|
|
(LINE, ("test_hotshot", g_lineno+1, "g")),
|
|
(ENTER, ("test_hotshot", f_lineno, "f")),
|
|
(LINE, ("test_hotshot", f_lineno, "f")),
|
|
(LINE, ("test_hotshot", f_lineno+1, "f")),
|
|
(LINE, ("test_hotshot", f_lineno+2, "f")),
|
|
(EXIT, ("test_hotshot", f_lineno, "f")),
|
|
(EXIT, ("test_hotshot", g_lineno, "g")),
|
|
]
|
|
self.run_test(g, events, self.new_profiler(lineevents=1))
|
|
|
|
|
|
def test_main():
|
|
test_support.run_unittest(HotShotTestCase)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|