Brendan Jackman | e81fdcb | 2017-01-04 17:10:29 +0000 | [diff] [blame] | 1 | # Copyright 2015-2017 ARM Limited |
Javi Merino | 034e7cc | 2015-04-22 18:39:21 +0100 | [diff] [blame] | 2 | # |
Javi Merino | aace7c0 | 2015-08-10 14:10:47 +0100 | [diff] [blame] | 3 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | # you may not use this file except in compliance with the License. |
| 5 | # You may obtain a copy of the License at |
| 6 | # |
| 7 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | # |
| 9 | # Unless required by applicable law or agreed to in writing, software |
| 10 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | # See the License for the specific language governing permissions and |
| 13 | # limitations under the License. |
| 14 | # |
| 15 | |
Javi Merino | 7f3c8a6 | 2014-06-13 11:38:58 +0100 | [diff] [blame] | 16 | |
| 17 | import unittest |
Javi Merino | 8e8e321 | 2014-06-26 15:24:34 +0100 | [diff] [blame] | 18 | import matplotlib |
| 19 | import pandas as pd |
Javi Merino | 7f3c8a6 | 2014-06-13 11:38:58 +0100 | [diff] [blame] | 20 | |
Javi Merino | 076277d | 2014-07-02 18:48:18 +0100 | [diff] [blame] | 21 | from test_thermal import BaseTestThermal |
Javi Merino | 435457c | 2015-08-10 15:59:10 +0100 | [diff] [blame] | 22 | import trappy |
Javi Merino | 7f3c8a6 | 2014-06-13 11:38:58 +0100 | [diff] [blame] | 23 | import plot_utils |
| 24 | |
| 25 | class TestPlotUtils(unittest.TestCase): |
Javi Merino | 7f3c8a6 | 2014-06-13 11:38:58 +0100 | [diff] [blame] | 26 | def test_normalize_title(self): |
| 27 | """Test normalize_title""" |
| 28 | self.assertEquals(plot_utils.normalize_title("Foo", ""), "Foo") |
| 29 | self.assertEquals(plot_utils.normalize_title("Foo", "Bar"), "Bar - Foo") |
Javi Merino | 3a73655 | 2014-06-19 19:22:44 +0100 | [diff] [blame] | 30 | |
Javi Merino | a156127 | 2014-06-21 17:49:02 +0100 | [diff] [blame] | 31 | def test_set_lim(self): |
| 32 | """Test set_lim()""" |
| 33 | |
| 34 | class GetSet(object): |
| 35 | def __init__(self): |
| 36 | self.min = 1 |
| 37 | self.max = 2 |
| 38 | |
| 39 | def get(self): |
| 40 | return (self.min, self.max) |
| 41 | |
| 42 | def set(self, minimum, maximum): |
| 43 | self.min = minimum |
| 44 | self.max = maximum |
| 45 | |
| 46 | gs = GetSet() |
| 47 | |
| 48 | plot_utils.set_lim("default", gs.get, gs.set) |
| 49 | self.assertEquals(gs.min, 1) |
| 50 | self.assertEquals(gs.max, 2) |
| 51 | |
| 52 | plot_utils.set_lim("range", gs.get, gs.set) |
| 53 | self.assertEquals(gs.min, 0.9) |
| 54 | self.assertEquals(gs.max, 2.1) |
| 55 | |
| 56 | plot_utils.set_lim((0, 100), gs.get, gs.set) |
| 57 | self.assertEquals(gs.min, 0) |
| 58 | self.assertEquals(gs.max, 100) |
| 59 | |
| 60 | def test_set_ylim(self): |
| 61 | """Test that set_ylim() doesn't bomb""" |
| 62 | |
Javi Merino | 5981ee8 | 2014-08-11 17:45:38 +0100 | [diff] [blame] | 63 | _, ax = matplotlib.pyplot.subplots() |
Javi Merino | a156127 | 2014-06-21 17:49:02 +0100 | [diff] [blame] | 64 | |
| 65 | plot_utils.set_ylim(ax, "default") |
| 66 | plot_utils.set_ylim(ax, (0, 5)) |
| 67 | |
| 68 | def test_set_xlim(self): |
| 69 | """Test that set_xlim() doesn't bomb""" |
| 70 | |
Javi Merino | 5981ee8 | 2014-08-11 17:45:38 +0100 | [diff] [blame] | 71 | _, ax = matplotlib.pyplot.subplots() |
Javi Merino | a156127 | 2014-06-21 17:49:02 +0100 | [diff] [blame] | 72 | |
| 73 | plot_utils.set_xlim(ax, "default") |
| 74 | plot_utils.set_xlim(ax, (0, 5)) |
| 75 | |
Javi Merino | f9ac578 | 2014-06-21 19:02:50 +0100 | [diff] [blame] | 76 | def test_pre_plot_setup(self): |
| 77 | """Test that plot_utils.pre_plot_setup() doesn't bomb""" |
| 78 | plot_utils.pre_plot_setup(None, None) |
| 79 | plot_utils.pre_plot_setup(height=9, width=None) |
| 80 | plot_utils.pre_plot_setup(height=None, width=9) |
| 81 | plot_utils.pre_plot_setup(3, 9) |
| 82 | |
Javi Merino | 7f88ae3 | 2014-08-11 16:53:35 +0100 | [diff] [blame] | 83 | axis = plot_utils.pre_plot_setup(ncols=2) |
| 84 | self.assertEquals(len(axis), 2) |
| 85 | |
Javi Merino | 270caac | 2014-08-12 10:41:38 +0100 | [diff] [blame] | 86 | axis = plot_utils.pre_plot_setup(nrows=2, ncols=3) |
| 87 | self.assertEquals(len(axis), 2) |
| 88 | self.assertEquals(len(axis[0]), 3) |
| 89 | self.assertEquals(len(axis[1]), 3) |
| 90 | |
Javi Merino | 3a73655 | 2014-06-19 19:22:44 +0100 | [diff] [blame] | 91 | def test_post_plot_setup(self): |
| 92 | """Test that post_plot_setup() doesn't bomb""" |
| 93 | |
Javi Merino | 5981ee8 | 2014-08-11 17:45:38 +0100 | [diff] [blame] | 94 | _, ax = matplotlib.pyplot.subplots() |
Javi Merino | 3a73655 | 2014-06-19 19:22:44 +0100 | [diff] [blame] | 95 | |
| 96 | plot_utils.post_plot_setup(ax) |
| 97 | plot_utils.post_plot_setup(ax, title="Foo") |
| 98 | plot_utils.post_plot_setup(ax, ylim=(0, 72)) |
Javi Merino | a156127 | 2014-06-21 17:49:02 +0100 | [diff] [blame] | 99 | plot_utils.post_plot_setup(ax, ylim="range") |
Javi Merino | f9d43af | 2014-06-21 16:32:04 +0100 | [diff] [blame] | 100 | plot_utils.post_plot_setup(ax, xlabel="Bar") |
Javi Merino | d04643f | 2014-06-21 17:00:16 +0100 | [diff] [blame] | 101 | plot_utils.post_plot_setup(ax, xlim=(0, 100)) |
Javi Merino | a156127 | 2014-06-21 17:49:02 +0100 | [diff] [blame] | 102 | plot_utils.post_plot_setup(ax, xlim="default") |
Javi Merino | 6a5c88d | 2014-06-19 09:43:18 +0100 | [diff] [blame] | 103 | |
Javi Merino | ed977c1 | 2014-06-25 17:46:17 +0100 | [diff] [blame] | 104 | def test_plot_hist(self): |
| 105 | """Test that plost_hist doesn't bomb""" |
| 106 | data = pd.Series([1, 1, 2, 4]) |
| 107 | |
Javi Merino | e5ea60a | 2014-08-12 16:41:42 +0100 | [diff] [blame] | 108 | _, ax = matplotlib.pyplot.subplots() |
Javi Merino | 23dff5d | 2015-03-09 19:05:46 +0000 | [diff] [blame] | 109 | plot_utils.plot_hist(data, ax, "Foo", "m", 20, "numbers", (0, 4), "default") |
Javi Merino | ed977c1 | 2014-06-25 17:46:17 +0100 | [diff] [blame] | 110 | |
Javi Merino | 076277d | 2014-07-02 18:48:18 +0100 | [diff] [blame] | 111 | class TestPlotUtilsNeedTrace(BaseTestThermal): |
Javi Merino | 6bf4883 | 2014-08-11 17:15:35 +0100 | [diff] [blame] | 112 | def __init__(self, *args, **kwargs): |
| 113 | super(TestPlotUtilsNeedTrace, self).__init__(*args, **kwargs) |
Javi Merino | d721b27 | 2015-04-20 15:37:53 +0100 | [diff] [blame] | 114 | self.map_label = {"00000000,00000039": "A53", "00000000,00000006": "A57"} |
| 115 | self.actor_order = ["GPU", "A57", "A53"] |
Javi Merino | 6bf4883 | 2014-08-11 17:15:35 +0100 | [diff] [blame] | 116 | |
Javi Merino | 9d9c9b8 | 2015-04-29 11:15:11 +0100 | [diff] [blame] | 117 | def test_number_freq_plots(self): |
| 118 | """Calculate the number of frequency plots correctly""" |
| 119 | trace_out = "" |
| 120 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 121 | trace = trappy.FTrace() |
| 122 | self.assertEquals(plot_utils.number_freq_plots([trace], self.map_label), |
Javi Merino | 9d9c9b8 | 2015-04-29 11:15:11 +0100 | [diff] [blame] | 123 | 3) |
| 124 | |
| 125 | # Strip out devfreq traces |
| 126 | with open("trace.txt") as fin: |
| 127 | for line in fin: |
| 128 | if ("thermal_power_devfreq_get_power:" not in line) and \ |
| 129 | ("thermal_power_devfreq_limit:" not in line): |
| 130 | trace_out += line |
| 131 | |
| 132 | with open("trace.txt", "w") as fout: |
| 133 | fout.write(trace_out) |
| 134 | |
| 135 | # Without devfreq there should only be two plots |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 136 | trace = trappy.FTrace() |
| 137 | self.assertEquals(plot_utils.number_freq_plots([trace], self.map_label), |
Javi Merino | 9d9c9b8 | 2015-04-29 11:15:11 +0100 | [diff] [blame] | 138 | 2) |
| 139 | |
Javi Merino | 2919e8d | 2014-06-26 15:16:05 +0100 | [diff] [blame] | 140 | def test_plot_temperature(self): |
| 141 | """Test that plot_utils.plot_temperature() doesn't bomb""" |
| 142 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 143 | trace1 = trappy.FTrace(name="first") |
| 144 | trace2 = trappy.FTrace(name="second") |
| 145 | traces = [trace1, trace2] |
Javi Merino | 2919e8d | 2014-06-26 15:16:05 +0100 | [diff] [blame] | 146 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 147 | plot_utils.plot_temperature(traces, ylim="default") |
Javi Merino | 8e8e321 | 2014-06-26 15:24:34 +0100 | [diff] [blame] | 148 | matplotlib.pyplot.close('all') |
Javi Merino | 6bf4883 | 2014-08-11 17:15:35 +0100 | [diff] [blame] | 149 | |
| 150 | def test_plot_load(self): |
| 151 | """Test that plot_utils.plot_load() doesn't bomb""" |
| 152 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 153 | trace1 = trappy.FTrace(name="first") |
| 154 | trace2 = trappy.FTrace(name="second") |
| 155 | traces = [trace1, trace2] |
Javi Merino | 6bf4883 | 2014-08-11 17:15:35 +0100 | [diff] [blame] | 156 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 157 | plot_utils.plot_load(traces, self.map_label, height=5) |
Javi Merino | 6bf4883 | 2014-08-11 17:15:35 +0100 | [diff] [blame] | 158 | matplotlib.pyplot.close('all') |
Javi Merino | af05f31 | 2014-08-11 17:47:59 +0100 | [diff] [blame] | 159 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 160 | def test_plot_load_single_trace(self): |
| 161 | """plot_utils.plot_load() can be used with a single trace""" |
| 162 | trace = trappy.FTrace() |
Javi Merino | af05f31 | 2014-08-11 17:47:59 +0100 | [diff] [blame] | 163 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 164 | plot_utils.plot_load([trace], self.map_label) |
Javi Merino | af05f31 | 2014-08-11 17:47:59 +0100 | [diff] [blame] | 165 | matplotlib.pyplot.close('all') |
Javi Merino | 0c48426 | 2014-08-12 11:55:04 +0100 | [diff] [blame] | 166 | |
| 167 | def test_plot_allfreqs(self): |
| 168 | """Test that plot_utils.plot_allfreqs() doesn't bomb""" |
| 169 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 170 | trace1 = trappy.FTrace(name="first") |
| 171 | trace2 = trappy.FTrace(name="second") |
| 172 | traces = [trace1, trace2] |
Javi Merino | 0c48426 | 2014-08-12 11:55:04 +0100 | [diff] [blame] | 173 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 174 | plot_utils.plot_allfreqs(traces, self.map_label, width=20) |
Javi Merino | 0c48426 | 2014-08-12 11:55:04 +0100 | [diff] [blame] | 175 | matplotlib.pyplot.close('all') |
| 176 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 177 | def test_plot_allfreqs_single_trace(self): |
| 178 | """plot_utils.plot_allfreqs() can be used with a single trace""" |
| 179 | trace = trappy.FTrace() |
Javi Merino | 114c9cb | 2014-08-12 14:47:09 +0100 | [diff] [blame] | 180 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 181 | plot_utils.plot_allfreqs([trace], self.map_label) |
Javi Merino | 0c48426 | 2014-08-12 11:55:04 +0100 | [diff] [blame] | 182 | matplotlib.pyplot.close('all') |
Javi Merino | 38fd12d | 2014-08-12 15:02:47 +0100 | [diff] [blame] | 183 | |
Javi Merino | 94d89c1 | 2016-01-05 14:47:07 +0000 | [diff] [blame] | 184 | def test_plot_allfreqs_one_actor(self): |
| 185 | """plot_utils.plot_allfreqs work when there is only one actor""" |
| 186 | |
| 187 | in_data = """ kworker/4:1-397 [004] 720.741349: thermal_power_cpu_get: cpus=00000000,00000006 freq=1400000 raw_cpu_power=189 load={23, 12} power=14 |
Javi Merino | 373c30f | 2016-03-08 13:59:25 +0000 | [diff] [blame] | 188 | kworker/4:1-397 [004] 720.741679: thermal_power_cpu_limit: cpus=00000000,00000006 freq=1400000 cdev_state=1 power=14""" |
Javi Merino | 94d89c1 | 2016-01-05 14:47:07 +0000 | [diff] [blame] | 189 | |
| 190 | with open("trace.txt", "w") as fout: |
| 191 | fout.write(in_data) |
| 192 | |
| 193 | traces = [trappy.FTrace(name="first"), trappy.FTrace(name="second")] |
| 194 | map_label = {"00000000,00000006": "A57"} |
| 195 | |
| 196 | plot_utils.plot_allfreqs(traces, map_label) |
| 197 | matplotlib.pyplot.close("all") |
| 198 | |
Javi Merino | 38fd12d | 2014-08-12 15:02:47 +0100 | [diff] [blame] | 199 | def test_plot_controller(self): |
| 200 | """plot_utils.plot_controller() doesn't bomb""" |
| 201 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 202 | trace1 = trappy.FTrace(name="first") |
| 203 | trace2 = trappy.FTrace(name="second") |
| 204 | traces = [trace1, trace2] |
Javi Merino | 38fd12d | 2014-08-12 15:02:47 +0100 | [diff] [blame] | 205 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 206 | plot_utils.plot_controller(traces, height=5) |
Javi Merino | 38fd12d | 2014-08-12 15:02:47 +0100 | [diff] [blame] | 207 | matplotlib.pyplot.close('all') |
Javi Merino | f5cd04b | 2014-08-12 15:26:22 +0100 | [diff] [blame] | 208 | |
| 209 | def test_plot_input_power(self): |
| 210 | """plot_utils.plot_input_power() doesn't bomb""" |
| 211 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 212 | trace1 = trappy.FTrace(name="first") |
| 213 | trace2 = trappy.FTrace(name="second") |
| 214 | traces = [trace1, trace2] |
Javi Merino | f5cd04b | 2014-08-12 15:26:22 +0100 | [diff] [blame] | 215 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 216 | plot_utils.plot_input_power(traces, self.actor_order, width=20) |
Javi Merino | f5cd04b | 2014-08-12 15:26:22 +0100 | [diff] [blame] | 217 | matplotlib.pyplot.close('all') |
Javi Merino | 9fde215 | 2014-08-12 15:34:24 +0100 | [diff] [blame] | 218 | |
| 219 | def test_plot_output_power(self): |
| 220 | """plot_utils.plot_output_power() doesn't bomb""" |
| 221 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 222 | trace1 = trappy.FTrace(name="first") |
| 223 | trace2 = trappy.FTrace(name="second") |
| 224 | traces = [trace1, trace2] |
Javi Merino | 9fde215 | 2014-08-12 15:34:24 +0100 | [diff] [blame] | 225 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 226 | plot_utils.plot_output_power(traces, self.actor_order, width=20) |
Javi Merino | 9fde215 | 2014-08-12 15:34:24 +0100 | [diff] [blame] | 227 | matplotlib.pyplot.close('all') |
Javi Merino | e5ea60a | 2014-08-12 16:41:42 +0100 | [diff] [blame] | 228 | |
| 229 | def test_plot_freq_hists(self): |
| 230 | """plot_utils.plot_freq_hists() doesn't bomb""" |
| 231 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 232 | trace1 = trappy.FTrace(name="first") |
| 233 | trace2 = trappy.FTrace(name="second") |
| 234 | traces = [trace1, trace2] |
Javi Merino | e5ea60a | 2014-08-12 16:41:42 +0100 | [diff] [blame] | 235 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 236 | plot_utils.plot_freq_hists(traces, self.map_label) |
Javi Merino | e5ea60a | 2014-08-12 16:41:42 +0100 | [diff] [blame] | 237 | matplotlib.pyplot.close('all') |
| 238 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 239 | def test_plot_freq_hists_single_trace(self): |
| 240 | """plot_utils.plot_freq_hists() works with a single trace""" |
Javi Merino | e5ea60a | 2014-08-12 16:41:42 +0100 | [diff] [blame] | 241 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 242 | trace = trappy.FTrace() |
Javi Merino | e5ea60a | 2014-08-12 16:41:42 +0100 | [diff] [blame] | 243 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 244 | plot_utils.plot_freq_hists([trace], self.map_label) |
Javi Merino | e5ea60a | 2014-08-12 16:41:42 +0100 | [diff] [blame] | 245 | matplotlib.pyplot.close('all') |
| 246 | |
| 247 | def test_plot_temperature_hist(self): |
| 248 | """plot_utils.plot_temperature_hist() doesn't bomb""" |
| 249 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 250 | trace1 = trappy.FTrace(name="first") |
| 251 | trace2 = trappy.FTrace(name="second") |
| 252 | traces = [trace1, trace2] |
Javi Merino | e5ea60a | 2014-08-12 16:41:42 +0100 | [diff] [blame] | 253 | |
Javi Merino | c26a323 | 2015-12-11 18:00:30 +0000 | [diff] [blame] | 254 | plot_utils.plot_temperature_hist(traces) |
Javi Merino | e5ea60a | 2014-08-12 16:41:42 +0100 | [diff] [blame] | 255 | matplotlib.pyplot.close('all') |