Commit 75adbb2c by Ben Patterson

Use pyinstrument instead.

parent 153ca29c
bok-choy==0.4.10 bok-choy==0.4.10
ddt==1.0.0 ddt==1.0.0
nose==1.3.7 nose==1.3.7
pyinstrument==0.13.1
...@@ -892,18 +892,29 @@ class FullWorkflowOverrideTest(OpenAssessmentTest, FullWorkflowMixin): ...@@ -892,18 +892,29 @@ class FullWorkflowOverrideTest(OpenAssessmentTest, FullWorkflowMixin):
""" """
def setUp(self): def setUp(self):
super(FullWorkflowOverrideTest, self).setUp("full_workflow_staff_override", staff=True) super(FullWorkflowOverrideTest, self).setUp("full_workflow_staff_override", staff=True)
import cProfile, pstats, StringIO # import cProfile, pstats, StringIO
pr = cProfile.Profile()
pr.enable() # pr = cProfile.Profile()
# pr.enable()
from pyinstrument import Profiler
self.profiler = Profiler()
self.profiler.start()
self.staff_area_page = StaffAreaPage(self.browser, self.problem_loc) self.staff_area_page = StaffAreaPage(self.browser, self.problem_loc)
self.addCleanup(pr.disable) # self.addCleanup(pr.disable)
# self.addCleanup(self.profiler.stop)
# self.addCleanup(self.log_to_file, profiler)
# self.addCleanup(self.break_fix, pr) # self.addCleanup(self.break_fix, pr)
self.addCleanup(pr.dump_stats, 'ben.log') # self.addCleanup(pr.dump_stats, 'ben.log')
def log_to_file(self):
with open('profile.log', 'w') as f:
f.write(self.profiler.output_text())
def log_to_file(pr): def tearDown(self):
with open('w', 'profile.log') as f: self.profiler.stop()
f.write(pr.print_stats()) self.log_to_file()
# def break_fix(self, pr): # def break_fix(self, pr):
# from nose.tools import set_trace # from nose.tools import set_trace
......
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