class TestTrace(TestCase):
"""Smoke tests for LISA's Trace class"""
traces_dir = os.path.join(os.path.dirname(__file__), 'traces')
events = [
'sched_switch',
]
def __init__(self, *args, **kwargs):
super(TestTrace, self).__init__(*args, **kwargs)
self.test_trace = os.path.join(self.traces_dir, 'test_trace.txt')
with open(os.path.join(self.traces_dir, 'platform.json')) as f:
self.platform = json.load(f)
trace_path = os.path.join(self.traces_dir, 'trace.txt')
self.trace = Trace(self.platform, trace_path, self.events)
def test_getTaskByName(self):
"""TestTrace: getTaskByName() returns the list of PIDs for all tasks with the specified name"""
for name, pids in [('watchdog/0', [12]),
('sh', [1642, 1702, 1717, 1718]),
('NOT_A_TASK', [])]:
self.assertEqual(self.trace.getTaskByName(name), pids)
def test_getTaskByPid(self):
"""TestTrace: getTaskByPid() returns the name of the task with the specified PID"""
for pid, names in [(15, 'watchdog/1'),
(1639, 'sshd'),
(987654321, None)]:
self.assertEqual(self.trace.getTaskByPid(pid), names)
def test_getTasks(self):
"""TestTrace: getTasks() returns a dictionary mapping PIDs to a single task name"""
tasks_dict = self.trace.getTasks()
for pid, name in [(1, 'init'),
(9, 'rcu_sched'),
(1383, 'jbd2/sda2-8')]:
self.assertEqual(tasks_dict[pid], name)
def test_setTaskName(self):
"""TestTrace: getTaskBy{Pid,Name}() properly track tasks renaming"""
in_data = """
father-1234 [002] 18765.018235: sched_switch: prev_comm=father prev_pid=1234 prev_prio=120 prev_state=0 next_comm=father next_pid=5678 next_prio=120
child-5678 [002] 18766.018236: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=sh next_pid=3367 next_prio=120
"""
with open(self.test_trace, "w") as fout:
fout.write(in_data)
trace = Trace(self.platform, self.test_trace, self.events)
self.assertEqual(trace.getTaskByPid(1234), 'father')
self.assertEqual(trace.getTaskByPid(5678), 'child')
self.assertEqual(trace.getTaskByName('father'), [1234])
os.remove(self.test_trace)
def test_setTaskName(self):
"""TestTrace: getTaskBy{Pid,Name}() properly track tasks renaming"""
in_data = """
father-1234 [002] 18765.018235: sched_switch: prev_comm=father prev_pid=1234 prev_prio=120 prev_state=0 next_comm=father next_pid=5678 next_prio=120
child-5678 [002] 18766.018236: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=sh next_pid=3367 next_prio=120
"""
with open(self.test_trace, "w") as fout:
fout.write(in_data)
trace = Trace(self.platform, self.test_trace, self.events)
self.assertEqual(trace.getTaskByPid(1234), 'father')
self.assertEqual(trace.getTaskByPid(5678), 'child')
self.assertEqual(trace.getTaskByName('father'), [1234])
os.remove(self.test_trace)
class TestTrace(TestCase):
"""Smoke tests for LISA's Trace class"""
traces_dir = os.path.join(os.path.dirname(__file__), 'traces')
events = [
'sched_switch', 'sched_overutilized', 'cpu_idle',
'sched_load_avg_task', 'sched_load_se'
]
def __init__(self, *args, **kwargs):
super(TestTrace, self).__init__(*args, **kwargs)
self.test_trace = os.path.join(self.traces_dir, 'test_trace.txt')
self.platform = self._get_platform()
self.trace_path = os.path.join(self.traces_dir, 'trace.txt')
self.trace = Trace(self.trace_path, self.events, self.platform)
def make_trace(self, in_data):
"""
Get a trace from an embedded string of textual trace data
"""
with open(self.test_trace, "w") as fout:
fout.write(in_data)
return Trace(self.test_trace,
self.events,
self.platform,
normalize_time=False)
def get_trace(self, trace_name):
"""
Get a trace from a separate provided trace file
"""
dir = os.path.join(self.traces_dir, trace_name)
trace_path = os.path.join(dir, 'trace.dat')
return Trace(trace_path, self.events, self._get_platform(trace_name))
def _get_platform(self, trace_name=None):
trace_dir = self.traces_dir
if trace_name:
trace_dir = os.path.join(trace_dir, trace_name)
with open(os.path.join(trace_dir, 'platform.json')) as f:
return json.load(f)
def test_getTaskByName(self):
"""TestTrace: getTaskByName() returns the list of PIDs for all tasks with the specified name"""
for name, pids in [('watchdog/0', [12]),
('sh', [1642, 1702, 1717, 1718]),
('NOT_A_TASK', [])]:
self.assertEqual(self.trace.getTaskByName(name), pids)
def test_getTaskByPid(self):
"""TestTrace: getTaskByPid() returns the name of the task with the specified PID"""
for pid, names in [(15, 'watchdog/1'), (1639, 'sshd'),
(987654321, None)]:
self.assertEqual(self.trace.getTaskByPid(pid), names)
def test_getTasks(self):
"""TestTrace: getTasks() returns a dictionary mapping PIDs to a single task name"""
tasks_dict = self.trace.getTasks()
for pid, name in [(1, 'init'), (9, 'rcu_sched'),
(1383, 'jbd2/sda2-8')]:
self.assertEqual(tasks_dict[pid], name)
def test_setTaskName(self):
"""TestTrace: getTaskBy{Pid,Name}() properly track tasks renaming"""
in_data = """
father-1234 [002] 18765.018235: sched_switch: prev_comm=father prev_pid=1234 prev_prio=120 prev_state=0 next_comm=father next_pid=5678 next_prio=120
child-5678 [002] 18766.018236: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=sh next_pid=3367 next_prio=120
"""
trace = self.make_trace(in_data)
self.assertEqual(trace.getTaskByPid(1234), 'father')
self.assertEqual(trace.getTaskByPid(5678), 'child')
self.assertEqual(trace.getTaskByName('father'), [1234])
os.remove(self.test_trace)
def test_time_range(self):
"""
TestTrace: time_range is the duration of the trace
"""
expected_duration = 6.676497
trace = Trace(self.trace_path,
self.events,
self.platform,
normalize_time=False)
self.assertAlmostEqual(trace.time_range, expected_duration, places=6)
def test_time_range_window(self):
"""
TestTrace: time_range is the duration of the trace in the given window
"""
expected_duration = 4.0
trace = Trace(self.trace_path,
self.events,
self.platform,
normalize_time=False,
window=(76.402065, 80.402065))
self.assertAlmostEqual(trace.time_range, expected_duration, places=6)
def test_overutilized_time(self):
"""
TestTrace: overutilized_time is the total time spent while system was overutilized
"""
events = [76.402065, 80.402065, 82.001337]
trace_end = self.trace.ftrace.basetime + self.trace.ftrace.get_duration(
)
# Last event should be extended to the trace's end
expected_time = (events[1] - events[0]) + (trace_end - events[2])
self.assertAlmostEqual(self.trace.overutilized_time,
expected_time,
places=6)
def test_plotCPUIdleStateResidency(self):
"""
Test that plotCPUIdleStateResidency doesn't crash
"""
in_data = """
foo-1 [000] 0.01: cpu_idle: state=0 cpu_id=0
foo-1 [000] 0.02: cpu_idle: state=-1 cpu_id=0
bar-2 [000] 0.03: cpu_idle: state=0 cpu_id=1
bar-2 [000] 0.04: cpu_idle: state=-1 cpu_id=1
baz-3 [000] 0.05: cpu_idle: state=0 cpu_id=2
baz-3 [000] 0.06: cpu_idle: state=-1 cpu_id=2
bam-4 [000] 0.07: cpu_idle: state=0 cpu_id=3
bam-4 [000] 0.08: cpu_idle: state=-1 cpu_id=3
child-5678 [002] 18765.018235: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=father next_pid=5678 next_prio=120
"""
trace = self.make_trace(in_data)
trace.analysis.idle.plotCPUIdleStateResidency()
def test_deriving_cpus_count(self):
"""Test that Trace derives cpus_count if it isn't provided"""
if self.platform:
del self.platform['cpus_count']
in_data = """
father-1234 [000] 18765.018235: sched_switch: prev_comm=father prev_pid=1234 prev_prio=120 prev_state=0 next_comm=father next_pid=5678 next_prio=120
child-5678 [002] 18765.018235: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=father next_pid=5678 next_prio=120
"""
trace = self.make_trace(in_data)
self.assertEqual(trace.platform['cpus_count'], 3)
def test_dfg_cpu_wakeups(self):
"""
Test the cpu_wakeups DataFrame getter
"""
trace = self.make_trace("""
<idle>-0 [004] 519.021928: cpu_idle: state=4294967295 cpu_id=4
<idle>-0 [004] 519.022147: cpu_idle: state=0 cpu_id=4
<idle>-0 [004] 519.022641: cpu_idle: state=4294967295 cpu_id=4
<idle>-0 [001] 519.022642: cpu_idle: state=4294967295 cpu_id=1
<idle>-0 [002] 519.022643: cpu_idle: state=4294967295 cpu_id=2
<idle>-0 [001] 519.022788: cpu_idle: state=0 cpu_id=1
<idle>-0 [002] 519.022831: cpu_idle: state=2 cpu_id=2
<idle>-0 [003] 519.022867: cpu_idle: state=4294967295 cpu_id=3
<idle>-0 [003] 519.023045: cpu_idle: state=2 cpu_id=3
<idle>-0 [004] 519.023080: cpu_idle: state=1 cpu_id=4
""")
df = trace.data_frame.cpu_wakeups()
exp_index = [
519.021928, 519.022641, 519.022642, 519.022643, 519.022867
]
exp_cpus = [4, 4, 1, 2, 3]
self.assertListEqual(df.index.tolist(), exp_index)
self.assertListEqual(df.cpu.tolist(), exp_cpus)
df = trace.data_frame.cpu_wakeups([2])
self.assertListEqual(df.index.tolist(), [519.022643])
self.assertListEqual(df.cpu.tolist(), [2])
def _test_tasks_dfs(self, trace_name):
"""Helper for smoke testing _dfg methods in tasks_analysis"""
trace = self.get_trace(trace_name)
lt_df = trace.data_frame.task_load_events()
columns = ['comm', 'pid', 'load_avg', 'util_avg', 'cpu']
if trace.has_big_little:
columns += ['cluster']
if 'nrg_model' in trace.platform:
columns += ['min_cluster_cap']
for column in columns:
msg = 'Task signals parsed from {} missing {} column'.format(
trace.data_dir, column)
self.assertIn(column, lt_df, msg=msg)
if trace.has_big_little:
df = trace.data_frame.top_big_tasks(min_samples=1)
for column in ['samples', 'comm']:
msg = 'Big tasks parsed from {} missing {} column'.format(
trace.data_dir, column)
self.assertIn(column, df, msg=msg)
# Pick an arbitrary PID to try plotting signals for.
pid = lt_df['pid'].unique()[0]
# Call plotTasks - although we won't check the results we can just check
# that things aren't totally borken.
trace.analysis.tasks.plotTasks(tasks=[pid])
def test_sched_load_signals(self):
"""Test parsing sched_load_se events from EAS upstream integration"""
self._test_tasks_dfs('sched_load')
def test_sched_load_avg_signals(self):
"""Test parsing sched_load_avg_task events from EAS1.2"""
self._test_tasks_dfs('sched_load_avg')
def test_getPeripheralClockInfo(self):
"""
TestTrace: getPeripheralClockInfo() returns proper effective rate info.
"""
self.events = ['clock_set_rate', 'clock_disable', 'clock_enable']
trace = self.make_trace("""
<idle>-0 [002] 380330000000: clock_enable: bus_clk state=1 cpu_id=2
<idle>-0 [002] 380331000000: clock_set_rate: bus_clk state=750000000 cpu_id=2
<idle>-0 [000] 380332000000: clock_disable: bus_clk state=0 cpu_id=0
<idle>-0 [000] 380333000000: clock_enable: bus_clk state=1 cpu_id=0
<idle>-0 [002] 380334000000: clock_set_rate: bus_clk state=100000000 cpu_id=2
<idle>-0 [000] 380335000000: clock_disable: bus_clk state=0 cpu_id=0
<idle>-0 [004] 380339000000: cpu_idle: state=1 cpu_id=4
""")
df = trace.getPeripheralClockEffectiveRate(clk_name='bus_clk')
exp_effective_rate = [
float('NaN'), 750000000, 0.0, 750000000, 100000000, 0.0
]
effective_rate = df['effective_rate'].tolist()
self.assertEqual(len(exp_effective_rate), len(effective_rate))
for e, r in zip(exp_effective_rate, effective_rate):
if (np.isnan(e)):
self.assertTrue(np.isnan(r))
continue
self.assertEqual(e, r)
class TestTrace(TestCase):
"""Smoke tests for LISA's Trace class"""
traces_dir = os.path.join(os.path.dirname(__file__), 'traces')
events = [
'sched_switch',
'sched_overutilized',
'cpu_idle',
'sched_load_avg_task',
'sched_load_se'
]
def __init__(self, *args, **kwargs):
super(TestTrace, self).__init__(*args, **kwargs)
self.test_trace = os.path.join(self.traces_dir, 'test_trace.txt')
self.platform = self._get_platform()
self.trace_path = os.path.join(self.traces_dir, 'trace.txt')
self.trace = Trace(self.platform, self.trace_path, self.events)
def make_trace(self, in_data):
"""
Get a trace from an embedded string of textual trace data
"""
with open(self.test_trace, "w") as fout:
fout.write(in_data)
return Trace(self.platform, self.test_trace, self.events,
normalize_time=False)
def get_trace(self, trace_name):
"""
Get a trace from a separate provided trace file
"""
dir = os.path.join(self.traces_dir, trace_name)
trace_path = os.path.join(dir, 'trace.dat')
return Trace(self._get_platform(trace_name), trace_path, self.events)
def _get_platform(self, trace_name=None):
trace_dir = self.traces_dir
if trace_name:
trace_dir = os.path.join(trace_dir, trace_name)
with open(os.path.join(trace_dir, 'platform.json')) as f:
return json.load(f)
def test_getTaskByName(self):
"""TestTrace: getTaskByName() returns the list of PIDs for all tasks with the specified name"""
for name, pids in [('watchdog/0', [12]),
('sh', [1642, 1702, 1717, 1718]),
('NOT_A_TASK', [])]:
self.assertEqual(self.trace.getTaskByName(name), pids)
def test_getTaskByPid(self):
"""TestTrace: getTaskByPid() returns the name of the task with the specified PID"""
for pid, names in [(15, 'watchdog/1'),
(1639, 'sshd'),
(987654321, None)]:
self.assertEqual(self.trace.getTaskByPid(pid), names)
def test_getTasks(self):
"""TestTrace: getTasks() returns a dictionary mapping PIDs to a single task name"""
tasks_dict = self.trace.getTasks()
for pid, name in [(1, 'init'),
(9, 'rcu_sched'),
(1383, 'jbd2/sda2-8')]:
self.assertEqual(tasks_dict[pid], name)
def test_setTaskName(self):
"""TestTrace: getTaskBy{Pid,Name}() properly track tasks renaming"""
in_data = """
father-1234 [002] 18765.018235: sched_switch: prev_comm=father prev_pid=1234 prev_prio=120 prev_state=0 next_comm=father next_pid=5678 next_prio=120
child-5678 [002] 18766.018236: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=sh next_pid=3367 next_prio=120
"""
trace = self.make_trace(in_data)
self.assertEqual(trace.getTaskByPid(1234), 'father')
self.assertEqual(trace.getTaskByPid(5678), 'child')
self.assertEqual(trace.getTaskByName('father'), [1234])
os.remove(self.test_trace)
def test_time_range(self):
"""
TestTrace: time_range is the duration of the trace
"""
expected_duration = 6.676497
trace = Trace(self.platform, self.trace_path,
self.events, normalize_time=False
)
self.assertAlmostEqual(trace.time_range, expected_duration, places=6)
def test_time_range_window(self):
"""
TestTrace: time_range is the duration of the trace in the given window
"""
expected_duration = 4.0
trace = Trace(self.platform, self.trace_path,
self.events, normalize_time=False,
window=(76.402065, 80.402065)
)
self.assertAlmostEqual(trace.time_range, expected_duration, places=6)
def test_overutilized_time(self):
"""
TestTrace: overutilized_time is the total time spent while system was overutilized
"""
events = [
76.402065,
80.402065,
82.001337
]
trace_end = self.trace.ftrace.basetime + self.trace.ftrace.get_duration()
# Last event should be extended to the trace's end
expected_time = (events[1] - events[0]) + (trace_end - events[2])
self.assertAlmostEqual(self.trace.overutilized_time, expected_time, places=6)
def test_plotCPUIdleStateResidency(self):
"""
Test that plotCPUIdleStateResidency doesn't crash
"""
in_data = """
foo-1 [000] 0.01: cpu_idle: state=0 cpu_id=0
foo-1 [000] 0.02: cpu_idle: state=-1 cpu_id=0
bar-2 [000] 0.03: cpu_idle: state=0 cpu_id=1
bar-2 [000] 0.04: cpu_idle: state=-1 cpu_id=1
baz-3 [000] 0.05: cpu_idle: state=0 cpu_id=2
baz-3 [000] 0.06: cpu_idle: state=-1 cpu_id=2
bam-4 [000] 0.07: cpu_idle: state=0 cpu_id=3
bam-4 [000] 0.08: cpu_idle: state=-1 cpu_id=3
child-5678 [002] 18765.018235: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=father next_pid=5678 next_prio=120
"""
trace = self.make_trace(in_data)
trace.analysis.idle.plotCPUIdleStateResidency()
def test_deriving_cpus_count(self):
"""Test that Trace derives cpus_count if it isn't provided"""
if self.platform:
del self.platform['cpus_count']
in_data = """
father-1234 [000] 18765.018235: sched_switch: prev_comm=father prev_pid=1234 prev_prio=120 prev_state=0 next_comm=father next_pid=5678 next_prio=120
child-5678 [002] 18765.018235: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=father next_pid=5678 next_prio=120
"""
trace = self.make_trace(in_data)
self.assertEqual(trace.platform['cpus_count'], 3)
def test_dfg_cpu_wakeups(self):
"""
Test the cpu_wakeups DataFrame getter
"""
trace = self.make_trace("""
<idle>-0 [004] 519.021928: cpu_idle: state=4294967295 cpu_id=4
<idle>-0 [004] 519.022147: cpu_idle: state=0 cpu_id=4
<idle>-0 [004] 519.022641: cpu_idle: state=4294967295 cpu_id=4
<idle>-0 [001] 519.022642: cpu_idle: state=4294967295 cpu_id=1
<idle>-0 [002] 519.022643: cpu_idle: state=4294967295 cpu_id=2
<idle>-0 [001] 519.022788: cpu_idle: state=0 cpu_id=1
<idle>-0 [002] 519.022831: cpu_idle: state=2 cpu_id=2
<idle>-0 [003] 519.022867: cpu_idle: state=4294967295 cpu_id=3
<idle>-0 [003] 519.023045: cpu_idle: state=2 cpu_id=3
<idle>-0 [004] 519.023080: cpu_idle: state=1 cpu_id=4
""")
df = trace.data_frame.cpu_wakeups()
exp_index=[519.021928, 519.022641, 519.022642, 519.022643, 519.022867]
exp_cpus= [ 4, 4, 1, 2, 3]
self.assertListEqual(df.index.tolist(), exp_index)
self.assertListEqual(df.cpu.tolist(), exp_cpus)
df = trace.data_frame.cpu_wakeups([2])
self.assertListEqual(df.index.tolist(), [519.022643])
self.assertListEqual(df.cpu.tolist(), [2])
def _test_tasks_dfs(self, trace_name):
"""Helper for smoke testing _dfg methods in tasks_analysis"""
trace = self.get_trace(trace_name)
lt_df = trace.data_frame.task_load_events()
columns = ['comm', 'pid', 'load_avg', 'util_avg', 'cpu']
if trace.has_big_little:
columns += ['cluster']
if 'nrg_model' in trace.platform:
columns += ['min_cluster_cap']
for column in columns:
msg = 'Task signals parsed from {} missing {} column'.format(
trace.data_dir, column)
self.assertIn(column, lt_df, msg=msg)
if trace.has_big_little:
df = trace.data_frame.top_big_tasks(min_samples=1)
for column in ['samples', 'comm']:
msg = 'Big tasks parsed from {} missing {} column'.format(
trace.data_dir, column)
self.assertIn(column, df, msg=msg)
# Pick an arbitrary PID to try plotting signals for.
pid = lt_df['pid'].unique()[0]
# Call plotTasks - although we won't check the results we can just check
# that things aren't totally borken.
trace.analysis.tasks.plotTasks(tasks=[pid])
def test_sched_load_signals(self):
"""Test parsing sched_load_se events from EAS upstream integration"""
self._test_tasks_dfs('sched_load')
def test_sched_load_avg_signals(self):
"""Test parsing sched_load_avg_task events from EAS1.2"""
self._test_tasks_dfs('sched_load_avg')
class TestTrace(TestCase):
"""Smoke tests for LISA's Trace class"""
traces_dir = os.path.join(os.path.dirname(__file__), 'traces')
events = [
'sched_switch',
'sched_overutilized',
'cpu_idle',
]
def __init__(self, *args, **kwargs):
super(TestTrace, self).__init__(*args, **kwargs)
self.test_trace = os.path.join(self.traces_dir, 'test_trace.txt')
self.platform = self._get_platform()
self.trace_path = os.path.join(self.traces_dir, 'trace.txt')
self.trace = Trace(self.platform, self.trace_path, self.events)
def make_trace(self, in_data):
with open(self.test_trace, "w") as fout:
fout.write(in_data)
return Trace(self.platform, self.test_trace, self.events,
normalize_time=False)
def _get_platform(self):
with open(os.path.join(self.traces_dir, 'platform.json')) as f:
return json.load(f)
def test_getTaskByName(self):
"""TestTrace: getTaskByName() returns the list of PIDs for all tasks with the specified name"""
for name, pids in [('watchdog/0', [12]),
('sh', [1642, 1702, 1717, 1718]),
('NOT_A_TASK', [])]:
self.assertEqual(self.trace.getTaskByName(name), pids)
def test_getTaskByPid(self):
"""TestTrace: getTaskByPid() returns the name of the task with the specified PID"""
for pid, names in [(15, 'watchdog/1'),
(1639, 'sshd'),
(987654321, None)]:
self.assertEqual(self.trace.getTaskByPid(pid), names)
def test_getTasks(self):
"""TestTrace: getTasks() returns a dictionary mapping PIDs to a single task name"""
tasks_dict = self.trace.getTasks()
for pid, name in [(1, 'init'),
(9, 'rcu_sched'),
(1383, 'jbd2/sda2-8')]:
self.assertEqual(tasks_dict[pid], name)
def test_setTaskName(self):
"""TestTrace: getTaskBy{Pid,Name}() properly track tasks renaming"""
in_data = """
father-1234 [002] 18765.018235: sched_switch: prev_comm=father prev_pid=1234 prev_prio=120 prev_state=0 next_comm=father next_pid=5678 next_prio=120
child-5678 [002] 18766.018236: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=sh next_pid=3367 next_prio=120
"""
trace = self.make_trace(in_data)
self.assertEqual(trace.getTaskByPid(1234), 'father')
self.assertEqual(trace.getTaskByPid(5678), 'child')
self.assertEqual(trace.getTaskByName('father'), [1234])
os.remove(self.test_trace)
def test_time_range(self):
"""
TestTrace: time_range is the duration of the trace
"""
expected_duration = 6.676497
trace = Trace(self.platform, self.trace_path,
self.events, normalize_time=False
)
self.assertAlmostEqual(trace.time_range, expected_duration, places=6)
def test_time_range_window(self):
"""
TestTrace: time_range is the duration of the trace in the given window
"""
expected_duration = 4.0
trace = Trace(self.platform, self.trace_path,
self.events, normalize_time=False,
window=(76.402065, 80.402065)
)
self.assertAlmostEqual(trace.time_range, expected_duration, places=6)
def test_overutilized_time(self):
"""
TestTrace: overutilized_time is the total time spent while system was overutilized
"""
events = [
76.402065,
80.402065,
82.001337
]
trace_end = self.trace.ftrace.basetime + self.trace.ftrace.get_duration()
# Last event should be extended to the trace's end
expected_time = (events[1] - events[0]) + (trace_end - events[2])
self.assertAlmostEqual(self.trace.overutilized_time, expected_time, places=6)
def test_plotCPUIdleStateResidency(self):
"""
Test that plotCPUIdleStateResidency doesn't crash
"""
in_data = """
foo-1 [000] 0.01: cpu_idle: state=0 cpu_id=0
foo-1 [000] 0.02: cpu_idle: state=-1 cpu_id=0
bar-2 [000] 0.03: cpu_idle: state=0 cpu_id=1
bar-2 [000] 0.04: cpu_idle: state=-1 cpu_id=1
baz-3 [000] 0.05: cpu_idle: state=0 cpu_id=2
baz-3 [000] 0.06: cpu_idle: state=-1 cpu_id=2
bam-4 [000] 0.07: cpu_idle: state=0 cpu_id=3
bam-4 [000] 0.08: cpu_idle: state=-1 cpu_id=3
child-5678 [002] 18765.018235: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=father next_pid=5678 next_prio=120
"""
trace = self.make_trace(in_data)
trace.analysis.idle.plotCPUIdleStateResidency()
def test_deriving_cpus_count(self):
"""Test that Trace derives cpus_count if it isn't provided"""
if self.platform:
del self.platform['cpus_count']
in_data = """
father-1234 [000] 18765.018235: sched_switch: prev_comm=father prev_pid=1234 prev_prio=120 prev_state=0 next_comm=father next_pid=5678 next_prio=120
child-5678 [002] 18765.018235: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=father next_pid=5678 next_prio=120
"""
trace = self.make_trace(in_data)
self.assertEqual(trace.platform['cpus_count'], 3)
def test_dfg_cpu_wakeups(self):
"""
Test the cpu_wakeups DataFrame getter
"""
trace = self.make_trace("""
<idle>-0 [004] 519.021928: cpu_idle: state=4294967295 cpu_id=4
<idle>-0 [004] 519.022147: cpu_idle: state=0 cpu_id=4
<idle>-0 [004] 519.022641: cpu_idle: state=4294967295 cpu_id=4
<idle>-0 [001] 519.022642: cpu_idle: state=4294967295 cpu_id=1
<idle>-0 [002] 519.022643: cpu_idle: state=4294967295 cpu_id=2
<idle>-0 [001] 519.022788: cpu_idle: state=0 cpu_id=1
<idle>-0 [002] 519.022831: cpu_idle: state=2 cpu_id=2
<idle>-0 [003] 519.022867: cpu_idle: state=4294967295 cpu_id=3
<idle>-0 [003] 519.023045: cpu_idle: state=2 cpu_id=3
<idle>-0 [004] 519.023080: cpu_idle: state=1 cpu_id=4
""")
df = trace.data_frame.cpu_wakeups()
exp_index=[519.021928, 519.022641, 519.022642, 519.022643, 519.022867]
exp_cpus= [ 4, 4, 1, 2, 3]
self.assertListEqual(df.index.tolist(), exp_index)
self.assertListEqual(df.cpu.tolist(), exp_cpus)
df = trace.data_frame.cpu_wakeups([2])
self.assertListEqual(df.index.tolist(), [519.022643])
self.assertListEqual(df.cpu.tolist(), [2])
class TestTrace(TestCase):
"""Smoke tests for LISA's Trace class"""
traces_dir = os.path.join(os.path.dirname(__file__), 'traces')
events = [
'sched_switch',
'sched_overutilized',
'cpu_idle',
'sched_load_avg_task',
'sched_load_se'
]
FLOAT_PLACES=6
def __init__(self, *args, **kwargs):
super(TestTrace, self).__init__(*args, **kwargs)
self.test_trace = os.path.join(self.traces_dir, 'test_trace.txt')
self.platform = self._get_platform()
self.trace_path = os.path.join(self.traces_dir, 'trace.txt')
self.trace = Trace(self.trace_path, self.events, self.platform)
def make_trace(self, in_data):
"""
Get a trace from an embedded string of textual trace data
"""
with open(self.test_trace, "w") as fout:
fout.write(in_data)
return Trace(self.test_trace, self.events,
self.platform, normalize_time=False)
def get_trace(self, trace_name):
"""
Get a trace from a separate provided trace file
"""
dir = os.path.join(self.traces_dir, trace_name)
trace_path = os.path.join(dir, 'trace.dat')
return Trace(trace_path, self.events,
self._get_platform(trace_name))
def _get_platform(self, trace_name=None):
trace_dir = self.traces_dir
if trace_name:
trace_dir = os.path.join(trace_dir, trace_name)
with open(os.path.join(trace_dir, 'platform.json')) as f:
return json.load(f)
def test_getTaskByName(self):
"""TestTrace: getTaskByName() returns the list of PIDs for all tasks with the specified name"""
for name, pids in [('watchdog/0', [12]),
('sh', [1642, 1702, 1717, 1718]),
('NOT_A_TASK', [])]:
self.assertEqual(self.trace.getTaskByName(name), pids)
def test_getTaskByPid(self):
"""TestTrace: getTaskByPid() returns the name of the task with the specified PID"""
for pid, names in [(15, 'watchdog/1'),
(1639, 'sshd'),
(987654321, None)]:
self.assertEqual(self.trace.getTaskByPid(pid), names)
def test_getTasks(self):
"""TestTrace: getTasks() returns a dictionary mapping PIDs to a single task name"""
tasks_dict = self.trace.getTasks()
for pid, name in [(1, 'init'),
(9, 'rcu_sched'),
(1383, 'jbd2/sda2-8')]:
self.assertEqual(tasks_dict[pid], name)
def test_setTaskName(self):
"""TestTrace: getTaskBy{Pid,Name}() properly track tasks renaming"""
in_data = """
father-1234 [002] 18765.018235: sched_switch: prev_comm=father prev_pid=1234 prev_prio=120 prev_state=0 next_comm=father next_pid=5678 next_prio=120
child-5678 [002] 18766.018236: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=sh next_pid=3367 next_prio=120
"""
trace = self.make_trace(in_data)
self.assertEqual(trace.getTaskByPid(1234), 'father')
self.assertEqual(trace.getTaskByPid(5678), 'child')
self.assertEqual(trace.getTaskByName('father'), [1234])
os.remove(self.test_trace)
def test_time_range(self):
"""
TestTrace: time_range is the duration of the trace
"""
expected_duration = 6.676497
trace = Trace(self.trace_path,
self.events,
self.platform,
normalize_time=False
)
self.assertAlmostEqual(trace.time_range, expected_duration,
places=self.FLOAT_PLACES)
def test_time_range_window(self):
"""
TestTrace: time_range is the duration of the trace in the given window
"""
expected_duration = 4.0
trace = Trace(self.trace_path,
self.events,
self.platform,
normalize_time=False,
window=(76.402065, 80.402065)
)
self.assertAlmostEqual(trace.time_range, expected_duration,
places=self.FLOAT_PLACES)
def test_squash_df(self):
"""
TestTrace: squash_df() behaves as expected
"""
index = [float(i) for i in range(15, 20)]
data = [(1, i % 2) for i in range(15, 20)]
df = pd.DataFrame(index=index, data=data, columns=['delta', 'state'])
## Test "standard" slice:
# The df here should be:
# Time delta state
# 16.5 .5 0
# 17 .5 1
df1 = Trace.squash_df(df, 16.5, 17.5,)
head = df1.head(1)
tail = df1.tail(1)
self.assertEquals(len(df1.index), 2)
self.assertEquals(df1.index.tolist(), [16.5, 17])
self.assertAlmostEqual(head['delta'].values[0], 0.5, places=self.FLOAT_PLACES)
self.assertAlmostEqual(tail['delta'].values[0], 0.5, places=self.FLOAT_PLACES)
self.assertEquals(head['state'].values[0], 0)
self.assertEquals(tail['state'].values[0], 1)
## Test slice where no event exists in the interval
# The df here should be:
# Time delta state
# 16.2 .6 0
df2 = Trace.squash_df(df, 16.2, 16.8)
self.assertEquals(len(df2.index), 1)
self.assertEquals(df2.index[0], 16.2)
self.assertAlmostEqual(df2['delta'].values[0], 0.6, places=self.FLOAT_PLACES)
self.assertEquals(df2['state'].values[0], 0)
## Test slice that matches an event's index
# The df here should be:
# Time delta state
# 16 1 0
df3 = Trace.squash_df(df, 16, 17)
self.assertEquals(len(df3.index), 1)
self.assertEquals(df3.index[0], 16)
self.assertAlmostEqual(df3['delta'].values[0], 1, places=self.FLOAT_PLACES)
self.assertEquals(df3['state'].values[0], 0)
## Test slice past last event
# The df here should be:
# Time delta state
# 19.5 .5 1
df4 = Trace.squash_df(df, 19.5, 22)
self.assertEquals(len(df4.index), 1)
self.assertEquals(df4.index[0], 19.5)
self.assertAlmostEqual(df4['delta'].values[0], 0.5, places=self.FLOAT_PLACES)
self.assertEquals(df4['state'].values[0], 1)
## Test slice where there's no past event
df5 = Trace.squash_df(df, 10, 30)
self.assertEquals(len(df5.index), 5)
## Test slice where that should contain nothing
df6 = Trace.squash_df(df, 8, 9)
self.assertEquals(len(df6.index), 0)
def test_overutilized_time(self):
"""
TestTrace: overutilized_time is the total time spent while system was overutilized
"""
events = [
76.402065,
80.402065,
82.001337
]
trace_end = self.trace.ftrace.basetime + self.trace.ftrace.get_duration()
# Last event should be extended to the trace's end
expected_time = (events[1] - events[0]) + (trace_end - events[2])
self.assertAlmostEqual(self.trace.overutilized_time, expected_time,
places=self.FLOAT_PLACES)
def test_plotCPUIdleStateResidency(self):
"""
Test that plotCPUIdleStateResidency doesn't crash
"""
in_data = """
foo-1 [000] 0.01: cpu_idle: state=0 cpu_id=0
foo-1 [000] 0.02: cpu_idle: state=-1 cpu_id=0
bar-2 [000] 0.03: cpu_idle: state=0 cpu_id=1
bar-2 [000] 0.04: cpu_idle: state=-1 cpu_id=1
baz-3 [000] 0.05: cpu_idle: state=0 cpu_id=2
baz-3 [000] 0.06: cpu_idle: state=-1 cpu_id=2
bam-4 [000] 0.07: cpu_idle: state=0 cpu_id=3
bam-4 [000] 0.08: cpu_idle: state=-1 cpu_id=3
child-5678 [002] 18765.018235: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=father next_pid=5678 next_prio=120
"""
trace = self.make_trace(in_data)
trace.analysis.idle.plotCPUIdleStateResidency()
def test_deriving_cpus_count(self):
"""Test that Trace derives cpus_count if it isn't provided"""
if self.platform:
del self.platform['cpus_count']
in_data = """
father-1234 [000] 18765.018235: sched_switch: prev_comm=father prev_pid=1234 prev_prio=120 prev_state=0 next_comm=father next_pid=5678 next_prio=120
child-5678 [002] 18765.018235: sched_switch: prev_comm=child prev_pid=5678 prev_prio=120 prev_state=1 next_comm=father next_pid=5678 next_prio=120
"""
trace = self.make_trace(in_data)
self.assertEqual(trace.platform['cpus_count'], 3)
def test_dfg_cpu_wakeups(self):
"""
Test the cpu_wakeups DataFrame getter
"""
trace = self.make_trace("""
<idle>-0 [004] 519.021928: cpu_idle: state=4294967295 cpu_id=4
<idle>-0 [004] 519.022147: cpu_idle: state=0 cpu_id=4
<idle>-0 [004] 519.022641: cpu_idle: state=4294967295 cpu_id=4
<idle>-0 [001] 519.022642: cpu_idle: state=4294967295 cpu_id=1
<idle>-0 [002] 519.022643: cpu_idle: state=4294967295 cpu_id=2
<idle>-0 [001] 519.022788: cpu_idle: state=0 cpu_id=1
<idle>-0 [002] 519.022831: cpu_idle: state=2 cpu_id=2
<idle>-0 [003] 519.022867: cpu_idle: state=4294967295 cpu_id=3
<idle>-0 [003] 519.023045: cpu_idle: state=2 cpu_id=3
<idle>-0 [004] 519.023080: cpu_idle: state=1 cpu_id=4
""")
df = trace.data_frame.cpu_wakeups()
exp_index=[519.021928, 519.022641, 519.022642, 519.022643, 519.022867]
exp_cpus= [ 4, 4, 1, 2, 3]
self.assertListEqual(df.index.tolist(), exp_index)
self.assertListEqual(df.cpu.tolist(), exp_cpus)
df = trace.data_frame.cpu_wakeups([2])
self.assertListEqual(df.index.tolist(), [519.022643])
self.assertListEqual(df.cpu.tolist(), [2])
def _test_tasks_dfs(self, trace_name):
"""Helper for smoke testing _dfg methods in tasks_analysis"""
trace = self.get_trace(trace_name)
lt_df = trace.data_frame.task_load_events()
columns = ['comm', 'pid', 'load_avg', 'util_avg', 'cpu']
if trace.has_big_little:
columns += ['cluster']
if 'nrg_model' in trace.platform:
columns += ['min_cluster_cap']
for column in columns:
msg = 'Task signals parsed from {} missing {} column'.format(
trace.data_dir, column)
self.assertIn(column, lt_df, msg=msg)
if trace.has_big_little:
df = trace.data_frame.top_big_tasks(min_samples=1)
for column in ['samples', 'comm']:
msg = 'Big tasks parsed from {} missing {} column'.format(
trace.data_dir, column)
self.assertIn(column, df, msg=msg)
# Pick an arbitrary PID to try plotting signals for.
pid = lt_df['pid'].unique()[0]
# Call plotTasks - although we won't check the results we can just check
# that things aren't totally borken.
trace.analysis.tasks.plotTasks(tasks=[pid])
def test_sched_load_signals(self):
"""Test parsing sched_load_se events from EAS upstream integration"""
self._test_tasks_dfs('sched_load')
def test_sched_load_avg_signals(self):
"""Test parsing sched_load_avg_task events from EAS1.2"""
self._test_tasks_dfs('sched_load_avg')
def test_getPeripheralClockInfo(self):
"""
TestTrace: getPeripheralClockInfo() returns proper effective rate info.
"""
self.events = [
'clock_set_rate',
'clock_disable',
'clock_enable'
]
trace = self.make_trace("""
<idle>-0 [002] 380330000000: clock_enable: bus_clk state=1 cpu_id=2
<idle>-0 [002] 380331000000: clock_set_rate: bus_clk state=750000000 cpu_id=2
<idle>-0 [000] 380332000000: clock_disable: bus_clk state=0 cpu_id=0
<idle>-0 [000] 380333000000: clock_enable: bus_clk state=1 cpu_id=0
<idle>-0 [002] 380334000000: clock_set_rate: bus_clk state=100000000 cpu_id=2
<idle>-0 [000] 380335000000: clock_disable: bus_clk state=0 cpu_id=0
<idle>-0 [004] 380339000000: cpu_idle: state=1 cpu_id=4
""")
df = trace.getPeripheralClockEffectiveRate(clk_name='bus_clk')
exp_effective_rate=[ float('NaN'), 750000000, 0.0, 750000000, 100000000, 0.0]
effective_rate = df['effective_rate'].tolist()
self.assertEqual(len(exp_effective_rate), len(effective_rate))
for e, r in zip(exp_effective_rate, effective_rate):
if (np.isnan(e)):
self.assertTrue(np.isnan(r))
continue
self.assertEqual(e,r)
