def test_merge_consecutive_self_time():
frame = Frame(
identifier='<module>\x00cibuildwheel/__init__.py\x0012',
children=[
Frame(identifier='strip_newlines\x00cibuildwheel/utils.py\x00997',
self_time=0.1),
SelfTimeFrame(self_time=0.2, ),
SelfTimeFrame(self_time=0.1, ),
Frame(identifier='calculate_metrics\x00cibuildwheel/utils.py\x007',
self_time=0.1),
SelfTimeFrame(self_time=0.05, ),
])
assert frame.time() == approx(0.55)
frame = processors.merge_consecutive_self_time(frame, options={})
assert frame.time() == approx(0.55)
assert len(frame.children) == 4
assert frame.children[0].self_time == approx(0.1)
assert frame.children[1].self_time == approx(0.3)
assert isinstance(frame.children[1], SelfTimeFrame)
assert frame.children[2].self_time == approx(0.1)
assert frame.children[3].self_time == approx(0.05)
assert isinstance(frame.children[3], SelfTimeFrame)
def test_remove_unnecessary_self_time_nodes():
frame = Frame(
identifier='<module>\x00cibuildwheel/__init__.py\x0012',
children=[
Frame(identifier='strip_newlines\x00cibuildwheel/utils.py\x00997',
children=[
SelfTimeFrame(self_time=0.2, ),
]),
SelfTimeFrame(self_time=0.5, ),
Frame(
identifier='strip_newlines\x00cibuildwheel/utils.py\x00997',
self_time=0.5,
),
Frame(
identifier='calculate_metrics\x00cibuildwheel/utils.py\x007',
self_time=0.1,
),
])
assert frame.time() == approx(1.3)
frame = processors.remove_unnecessary_self_time_nodes(frame, options={})
assert frame.time() == approx(1.3)
assert len(frame.children) == 4
# check the self time node was deleted
strip_newlines_frame = frame.children[0]
assert strip_newlines_frame.function == 'strip_newlines'
assert len(strip_newlines_frame.children) == 0
assert strip_newlines_frame.self_time == 0.2
def test_remove_irrelevant_nodes():
frame = Frame(
identifier='<module>\x00cibuildwheel/__init__.py\x0012',
children=[
Frame(
identifier='strip_newlines\x00cibuildwheel/utils.py\x00997',
children=[
Frame(
identifier='scan_string\x00cibuildwheel/utils.py\x0054',
self_time=10,
),
]),
SelfTimeFrame(self_time=0.5, ),
Frame(
identifier='strip_newlines\x00cibuildwheel/utils.py\x00997',
self_time=0.5,
),
Frame(
identifier='calculate_metrics\x00cibuildwheel/utils.py\x007',
self_time=0.01,
),
])
assert frame.time() == approx(11.01)
frame = processors.remove_irrelevant_nodes(frame, options={})
assert frame.time() == approx(11.01)
# check the calculate metrics function was deleted
assert len(frame.children) == 3
assert 'calculate_metrics' not in [f.function for f in frame.children]
def test_group_library_frames_processor():
frame = Frame(
identifier='<module>\x00cibuildwheel/__init__.py\x0012',
children=[
Frame(
identifier=
'library_function\x00env/lib/python3.6/django/__init__.py\x00997',
children=[
Frame(
identifier=
'library_inner\x00env/lib/python3.6/django/http.py\x0054',
children=[
Frame(
identifier=
'library_callback\x00env/lib/python3.6/django/views.py\x0054',
children=[
Frame(
identifier=
'<module>\x00cibuildwheel/views.py\x0012',
self_time=0.3,
),
]),
]),
]),
SelfTimeFrame(self_time=0.5, ),
Frame(
identifier='strip_newlines\x00cibuildwheel/utils.py\x00997',
self_time=0.5,
),
Frame(
identifier='calculate_metrics\x00cibuildwheel/utils.py\x007',
self_time=0.1,
),
])
assert frame.time() == approx(1.4)
frame = processors.group_library_frames_processor(frame, options={})
assert frame.time() == approx(1.4)
group_root = frame.children[0]
group = group_root.group
assert group.root == group_root
for frame in group.frames:
assert frame.group == group
assert group_root in group.frames
assert group_root.children[0] in group.frames
assert group_root.children[0].children[0] in group.frames
assert group_root.children[0].children[0] in group.exit_frames
assert group_root.children[0].children[0].children[0] not in group.frames
old_sys_path = sys.path[:]
sys.path.append('env/lib/python3.6')
assert group.libraries == ['django']
sys.path[:] = old_sys_path
def to_frame(agg):
if agg.seq is None:
identifier = agg.identifier
else:
function, filename, line = agg.identifier.split('\x00')
identifier = f"{function}:{agg.seq}\x00{filename}\x00{line}"
frame = Frame(identifier)
frame.add_children([to_frame(ch) for ch in agg.children.values()])
frame.add_child(SelfTimeFrame(self_time=agg.self_time))
return frame
def test_aggregate_repeated_calls():
frame = Frame(
identifier='<module>\x00cibuildwheel/__init__.py\x0012',
children=[
Frame(
identifier='strip_newlines\x00cibuildwheel/utils.py\x00997',
self_time=0.1,
children=[
Frame(
identifier='scan_string\x00cibuildwheel/utils.py\x0054',
self_time=0.2,
),
]),
SelfTimeFrame(self_time=0.1, ),
Frame(
identifier='strip_newlines\x00cibuildwheel/utils.py\x00997',
self_time=0.1,
),
SelfTimeFrame(self_time=0.2, ),
Frame(
identifier='calculate_metrics\x00cibuildwheel/utils.py\x007',
self_time=0.1,
),
SelfTimeFrame(self_time=0.05, ),
])
assert frame.time() == approx(0.85)
frame = processors.aggregate_repeated_calls(frame, options={})
assert frame.time() == approx(0.85)
# children should be sorted by time
assert len(frame.children) == 3
assert frame.children[0].function == 'strip_newlines'
assert frame.children[0].time() == 0.4
assert frame.children[0].children[0].function == 'scan_string'
assert isinstance(frame.children[1], SelfTimeFrame)
assert frame.children[1].time() == approx(0.35)
def root_frame(self, trim_stem=True):
'''
Parses the internal frame records and returns a tree of Frame objects
'''
root_frame = None
frame_stack = []
for frame_tuple in self.frame_records:
identifier_stack = frame_tuple[0]
time = frame_tuple[1]
# now we must create a stack of frame objects and assign this time to the leaf
for stack_depth, frame_identifier in enumerate(identifier_stack):
if stack_depth < len(frame_stack):
if frame_identifier != frame_stack[stack_depth].identifier:
# trim any frames after and including this one
del frame_stack[stack_depth:]
if stack_depth >= len(frame_stack):
frame = Frame(frame_identifier)
frame_stack.append(frame)
if stack_depth == 0:
# There should only be one root frame, as far as I know
assert root_frame is None, ASSERTION_MESSAGE
root_frame = frame
else:
parent = frame_stack[stack_depth - 1]
parent.add_child(frame)
# trim any extra frames
del frame_stack[stack_depth + 1:] # pylint: disable=W0631
# assign the time to the final frame
frame_stack[-1].add_child(SelfTimeFrame(self_time=time))
if root_frame is None:
return None
if trim_stem:
root_frame = self._trim_stem(root_frame)
return root_frame