def test_profile(self):
def g():
pass
def f():
g()
def main():
f()
f()
f()
with tracebin.record() as recorder:
main()
assert recorder.calls is None
start = time.time()
with tracebin.record(profile=True) as recorder:
main()
assert len(recorder.calls) == 3
# There are calls to high_res_time() before and after main, in order to
# record the total execution time
[call1, call2, call3] = recorder.calls
assert call1.func_name == "high_res_time"
assert len(call1.subcalls) == 1
assert call2.func_name == "main"
assert (call2.start_time - start) < 1
assert (call2.end_time - call2.start_time) < 1
assert len(call2.subcalls) == 3
assert {c.func_name for c in call2.subcalls} == {"f"}
assert call2.subcalls[1].subcalls[0].func_name == "g"
def test_profile(self):
def g():
pass
def f():
g()
def main():
f()
f()
f()
with tracebin.record() as recorder:
main()
assert recorder.calls is None
start = time.time()
with tracebin.record(profile=True) as recorder:
main()
assert len(recorder.calls) == 3
# There are calls to high_res_time() before and after main, in order to
# record the total execution time
[call1, call2, call3] = recorder.calls
assert call1.func_name == "high_res_time"
assert len(call1.subcalls) == 1
assert call2.func_name == "main"
assert (call2.start_time - start) < 1
assert (call2.end_time - call2.start_time) < 1
assert len(call2.subcalls) == 3
assert {c.func_name for c in call2.subcalls} == {"f"}
assert call2.subcalls[1].subcalls[0].func_name == "g"
def test_record_stdout(self):
with tracebin.record() as recorder:
pass
assert recorder.stdout == ""
assert recorder.stderr == ""
with tracebin.record() as recorder:
print(None)
assert recorder.stdout == "None\n"
assert recorder.stderr == ""
with tracebin.record() as recorder:
print(None, file=sys.stderr)
assert recorder.stdout == ""
assert recorder.stderr == "None\n"
def test_record_stdout(self):
with tracebin.record() as recorder:
pass
assert recorder.stdout == ""
assert recorder.stderr == ""
with tracebin.record() as recorder:
print(None)
assert recorder.stdout == "None\n"
assert recorder.stderr == ""
with tracebin.record() as recorder:
print(None, file=sys.stderr)
assert recorder.stdout == ""
assert recorder.stderr == "None\n"
def test_inlined_function(self):
def f(i):
return i - 1
def main():
i = 1500
while i > 0:
i = f(i)
i
with tracebin.record() as recorder:
main()
[trace] = recorder.traces
[entry, preamble, loop] = trace.sections
assert len(loop.chunks) == 10
assert loop.chunks[0].get_op_names() == ["label"]
assert loop.chunks[1].sourcelines == [
""" def main():\n""",
""" i = 1500\n""",
""" while i > 0:\n""",
]
assert loop.chunks[2].get_op_names() == [
"debug_merge_point", "debug_merge_point", "debug_merge_point",
"int_gt", "guard_true", "debug_merge_point"
]
assert loop.chunks[3].sourcelines == [
""" i = f(i)\n""",
]
assert loop.chunks[4].get_op_names() == [
"debug_merge_point", "debug_merge_point", "debug_merge_point",
"force_token"
]
assert loop.chunks[5].sourcelines == [
""" def f(i):\n""",
""" return i - 1\n""",
]
assert loop.chunks[5].linenos == [122, 123]
assert loop.chunks[6].get_op_names() == [
"debug_merge_point", "debug_merge_point", "debug_merge_point",
"int_sub", "debug_merge_point"
]
assert loop.chunks[7].get_op_names() == ["debug_merge_point"]
assert loop.chunks[8].sourcelines == [
""" i\n""",
]
if self.thread_spawned():
assert loop.chunks[9].get_op_names() == [
"debug_merge_point", "debug_merge_point", "debug_merge_point",
"guard_not_invalidated", "getfield_raw", "int_sub",
"setfield_raw", "int_lt", "guard_false", "debug_merge_point",
"jump"
]
else:
assert loop.chunks[9].get_op_names() == [
"debug_merge_point", "debug_merge_point", "debug_merge_point",
"guard_not_invalidated", "getfield_raw", "int_lt",
"guard_false", "debug_merge_point", "jump"
]
def test_trace(self, serializer_cls):
with tracebin.record() as recorder:
for i in xrange(1500):
pass
serializer = serializer_cls(recorder)
serializer.dump()
def test_trace_profilehook(self):
def profile(frame, event, arg):
pass
def sub(x, y):
return x - y
def main():
i = 1500
orig_profile = sys.getprofile()
sys.setprofile(profile)
try:
while i > 0:
i = sub(i, 1)
finally:
sys.setprofile(orig_profile)
with tracebin.record() as recorder:
main()
[trace] = recorder.traces
[_, _, loop] = trace.sections
py_profile_chunk = loop.chunks[9]
assert py_profile_chunk.sourcelines == [
""" def profile(frame, event, arg):\n""",
""" pass\n""",
]
op_profile_chunk = loop.chunks[10]
assert op_profile_chunk.get_op_names() == ["debug_merge_point", "debug_merge_point"]
def test_abort(self, serializer_cls):
with tracebin.record() as recorder:
for i in xrange(1500):
sys.exc_info()
serializer = serializer_cls(recorder)
serializer.dump()
def test_trace_profilehook(self):
def profile(frame, event, arg):
pass
def sub(x, y):
return x - y
def main():
i = 1500
orig_profile = sys.getprofile()
sys.setprofile(profile)
try:
while i > 0:
i = sub(i, 1)
finally:
sys.setprofile(orig_profile)
with tracebin.record() as recorder:
main()
[trace] = recorder.traces
[_, _, loop] = trace.sections
py_profile_chunk = loop.chunks[9]
assert py_profile_chunk.sourcelines == [
""" def profile(frame, event, arg):\n""",
""" pass\n""",
]
op_profile_chunk = loop.chunks[10]
assert op_profile_chunk.get_op_names() == [
"debug_merge_point", "debug_merge_point"
]
def test_record_build_options(self):
with tracebin.record() as recorder:
pass
assert recorder.options["build"]["gc"] == sys.pypy_translation_info["translation.gc"]
assert recorder.options["build"]["gcrootfinder"] == sys.pypy_translation_info["translation.gcrootfinder"]
assert recorder.options["build"]["pypy_version"] == sys._mercurial[2]
def test_record_build_options(self):
with tracebin.record() as recorder:
pass
assert recorder.options["build"]["gc"] == sys.pypy_translation_info[
"translation.gc"]
assert recorder.options["build"][
"gcrootfinder"] == sys.pypy_translation_info[
"translation.gcrootfinder"]
assert recorder.options["build"]["pypy_version"] == sys._mercurial[2]
def test_recorder(self, serializer_cls):
with tracebin.record() as recorder:
pass
serializer = serializer_cls(recorder)
data = serializer.get_data()
assert data.viewkeys() == {"traces", "aborts", "runtime", "stdout", "stderr", "options", "calls"}
assert data["calls"] is None
dump = serializer.dump()
assert data == serializer_cls.load(dump)
def test_profile_builtin_exception(self):
import math
with tracebin.record(profile=True) as recorder:
try:
math.sqrt("a")
except TypeError:
pass
[_, call, _] = recorder.calls
assert call.func_name == "sqrt"
assert call.subcalls == []
def test_profile_builtin_exception(self):
import math
with tracebin.record(profile=True) as recorder:
try:
math.sqrt("a")
except TypeError:
pass
[_, call, _] = recorder.calls
assert call.func_name == "sqrt"
assert call.subcalls == []
def test_record_aborts(self):
def f():
for i in xrange(1500):
sys.exc_info()
with tracebin.record() as recorder:
f()
assert len(recorder.aborts) == 1
[abort] = recorder.aborts
assert abort.reason == "ABORT_ESCAPE"
assert abort.pycode is f.__code__
# If this fails check to make sure it still matches the line
# `sys.exc_info`
assert abort.lineno == 72
def test_record_loop(self):
def f(n):
while n > 0:
n -= 1
with tracebin.record() as recorder:
f(1500)
assert len(recorder.traces) == 1
[trace] = recorder.traces
assert trace.root_file == __file__
assert trace.root_function == "f"
assert len(trace.sections) == 3
assert [s.label for s in trace.sections] == ["Entry", "Preamble", "Loop body"]
def test_record_loop(self):
def f(n):
while n > 0:
n -= 1
with tracebin.record() as recorder:
f(1500)
assert len(recorder.traces) == 1
[trace] = recorder.traces
assert trace.root_file == __file__
assert trace.root_function == "f"
assert len(trace.sections) == 3
assert [s.label
for s in trace.sections] == ["Entry", "Preamble", "Loop body"]
def test_record_aborts(self):
def f():
for i in xrange(1500):
sys.exc_info()
with tracebin.record() as recorder:
f()
assert len(recorder.aborts) == 1
[abort] = recorder.aborts
assert abort.reason == "ABORT_ESCAPE"
assert abort.pycode is f.__code__
# If this fails check to make sure it still matches the line
# `sys.exc_info`
assert abort.lineno == 72
def test_code_interpolation(self):
# This test is pretty fragile, but I don't see how else to make sure we
# don't screw this up.
def f():
i = 1500
while i > 0:
i -= 1
i
with tracebin.record() as recorder:
f()
[trace] = recorder.traces
[entry, preamble, loop] = trace.sections
assert len(loop.chunks) == 7
assert loop.chunks[0].get_op_names() == ["label"]
assert loop.chunks[1].sourcelines == [
""" def f():\n""", """ i = 1500\n""",
""" while i > 0:\n"""
]
assert loop.chunks[1].linenos == [87, 88, 89]
assert loop.chunks[2].get_op_names() == [
"debug_merge_point", "debug_merge_point", "debug_merge_point",
"int_gt", "guard_true", "debug_merge_point"
]
assert loop.chunks[3].sourcelines == [""" i -= 1\n"""]
assert loop.chunks[3].linenos == [90]
assert loop.chunks[4].get_op_names() == [
"debug_merge_point", "debug_merge_point", "debug_merge_point",
"int_sub", "debug_merge_point"
]
assert loop.chunks[5].sourcelines == [""" i\n"""]
if self.thread_spawned():
assert loop.chunks[6].get_op_names() == [
"debug_merge_point", "debug_merge_point", "debug_merge_point",
"guard_not_invalidated", "getfield_raw", "int_sub",
"setfield_raw", "int_lt", "guard_false", "debug_merge_point",
"jump"
]
else:
assert loop.chunks[6].get_op_names() == [
"debug_merge_point", "debug_merge_point", "debug_merge_point",
"guard_not_invalidated", "getfield_raw", "int_lt",
"guard_false", "debug_merge_point", "jump"
]
def test_calls(self, serializer_cls):
def f():
pass
def main():
f()
f()
f()
with tracebin.record(profile=True) as recorder:
main()
serializer = serializer_cls(recorder)
dump = serializer.dump()
data = serializer.load(dump)
assert len(data["calls"]) == 3
assert len(data["calls"][1]["subcalls"]) == 3
def test_profile_exception(self):
def f():
raise ValueError
def main():
try:
f()
except ValueError:
pass
with tracebin.record(profile=True) as recorder:
main()
[_, call, _] = recorder.calls
assert call.func_name == "main"
[subcall] = call.subcalls
assert subcall.func_name == "f"
def test_profile_exception(self):
def f():
raise ValueError
def main():
try:
f()
except ValueError:
pass
with tracebin.record(profile=True) as recorder:
main()
[_, call, _] = recorder.calls
assert call.func_name == "main"
[subcall] = call.subcalls
assert subcall.func_name == "f"
def test_inlined_function(self):
def f(i):
return i - 1
def main():
i = 1500
while i > 0:
i = f(i)
i
with tracebin.record() as recorder:
main()
[trace] = recorder.traces
[entry, preamble, loop] = trace.sections
assert len(loop.chunks) == 10
assert loop.chunks[0].get_op_names() == ["label"]
assert loop.chunks[1].sourcelines == [
""" def main():\n""",
""" i = 1500\n""",
""" while i > 0:\n""",
]
assert loop.chunks[2].get_op_names() == ["debug_merge_point", "debug_merge_point", "debug_merge_point", "int_gt", "guard_true", "debug_merge_point"]
assert loop.chunks[3].sourcelines == [
""" i = f(i)\n""",
]
assert loop.chunks[4].get_op_names() == ["debug_merge_point", "debug_merge_point", "debug_merge_point", "force_token"]
assert loop.chunks[5].sourcelines == [
""" def f(i):\n""",
""" return i - 1\n""",
]
assert loop.chunks[5].linenos == [122, 123]
assert loop.chunks[6].get_op_names() == ["debug_merge_point", "debug_merge_point", "debug_merge_point", "int_sub", "debug_merge_point"]
assert loop.chunks[7].get_op_names() == ["debug_merge_point"]
assert loop.chunks[8].sourcelines == [
""" i\n""",
]
if self.thread_spawned():
assert loop.chunks[9].get_op_names() == ["debug_merge_point", "debug_merge_point", "debug_merge_point", "guard_not_invalidated", "getfield_raw", "int_sub", "setfield_raw", "int_lt", "guard_false", "debug_merge_point", "jump"]
else:
assert loop.chunks[9].get_op_names() == ["debug_merge_point", "debug_merge_point", "debug_merge_point", "guard_not_invalidated", "getfield_raw", "int_lt", "guard_false", "debug_merge_point", "jump"]
def test_code_interpolation(self):
# This test is pretty fragile, but I don't see how else to make sure we
# don't screw this up.
def f():
i = 1500
while i > 0:
i -= 1
i
with tracebin.record() as recorder:
f()
[trace] = recorder.traces
[entry, preamble, loop] = trace.sections
assert len(loop.chunks) == 7
assert loop.chunks[0].get_op_names() == ["label"]
assert loop.chunks[1].sourcelines == [
""" def f():\n""",
""" i = 1500\n""",
""" while i > 0:\n"""
]
assert loop.chunks[1].linenos == [87, 88, 89]
assert loop.chunks[2].get_op_names() == ["debug_merge_point", "debug_merge_point", "debug_merge_point", "int_gt", "guard_true", "debug_merge_point"]
assert loop.chunks[3].sourcelines == [
""" i -= 1\n"""
]
assert loop.chunks[3].linenos == [90]
assert loop.chunks[4].get_op_names() == ["debug_merge_point", "debug_merge_point", "debug_merge_point", "int_sub", "debug_merge_point"]
assert loop.chunks[5].sourcelines == [
""" i\n"""
]
if self.thread_spawned():
assert loop.chunks[6].get_op_names() == ["debug_merge_point", "debug_merge_point", "debug_merge_point", "guard_not_invalidated", "getfield_raw", "int_sub", "setfield_raw", "int_lt", "guard_false", "debug_merge_point", "jump"]
else:
assert loop.chunks[6].get_op_names() == ["debug_merge_point", "debug_merge_point", "debug_merge_point", "guard_not_invalidated", "getfield_raw", "int_lt", "guard_false", "debug_merge_point", "jump"]
def test_record_runtime(self):
with tracebin.record() as recorder:
pass
assert recorder.runtime < 1
def test_record_jit_options(self):
with tracebin.record() as recorder:
pass
assert recorder.options["jit"].viewkeys() == pypyjit.defaults.viewkeys()
def test_record_jit_options(self):
with tracebin.record() as recorder:
pass
assert recorder.options["jit"].viewkeys() == pypyjit.defaults.viewkeys(
)
def test_record_runtime(self):
with tracebin.record() as recorder:
pass
assert recorder.runtime < 1
