1 # Copyright (c) 2022 Cisco and/or its affiliates.
2 # Licensed under the Apache License, Version 2.0 (the "License");
3 # you may not use this file except in compliance with the License.
4 # You may obtain a copy of the License at:
6 # http://www.apache.org/licenses/LICENSE-2.0
8 # Unless required by applicable law or agreed to in writing, software
9 # distributed under the License is distributed on an "AS IS" BASIS,
10 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11 # See the License for the specific language governing permissions and
12 # limitations under the License.
17 import plotly.graph_objects as go
21 def select_data(data: pd.DataFrame, itm:str) -> pd.DataFrame:
22 """Select the data for graphs from the provided data frame.
24 :param data: Data frame with data for graphs.
25 :param itm: Item (in this case job name) which data will be selected from
27 :type data: pandas.DataFrame
29 :returns: A data frame with selected data.
30 :rtype: pandas.DataFrame
33 df = data.loc[(data["job"] == itm)].sort_values(
34 by="start_time", ignore_index=True)
35 df = df.dropna(subset=["duration", ])
40 def graph_statistics(df: pd.DataFrame, job:str, layout: dict) -> tuple:
42 1. Passed / failed tests,
44 with additional information shown in hover.
46 :param df: Data frame with input data.
47 :param job: The name of job which data will be presented in the graphs.
48 :param layout: Layout of plot.ly graph.
49 :type df: pandas.DataFrame
52 :returns: Tuple with two generated graphs (pased/failed tests and job
54 :rtype: tuple(plotly.graph_objects.Figure, plotly.graph_objects.Figure)
57 data = select_data(df, job)
62 for _, row in data.iterrows():
63 d_type = "trex" if row["dut_type"] == "none" else row["dut_type"]
65 f"date: {row['start_time'].strftime('%Y-%m-%d %H:%M:%S')}<br>"
67 f"{(int(row['duration']) // 3600):02d}:"
68 f"{((int(row['duration']) % 3600) // 60):02d}<br>"
69 f"passed: {row['passed']}<br>"
70 f"failed: {row['failed']}<br>"
71 f"{d_type}-ref: {row['dut_version']}<br>"
72 f"csit-ref: {row['job']}/{row['build']}<br>"
73 f"hosts: {', '.join(row['hosts'])}"
75 hover.append(hover_itm)
78 fig_duration = go.Figure(
89 step = max(data["duration"]) / 5
91 tickvals.append(int(step * (i + 1)))
92 layout_duration = layout.get("plot-stats-duration", dict())
94 layout_duration["yaxis"]["tickvals"] = tickvals
95 layout_duration["yaxis"]["ticktext"] = [
96 f"{(val // 3600):02d}:{((val % 3600) // 60):02d}" \
99 fig_duration.update_layout(layout_duration)
102 fig_passed = go.Figure(
105 x=data["start_time"],
112 x=data["start_time"],
120 layout_pf = layout.get("plot-stats-passed", dict())
122 fig_passed.update_layout(layout_pf)
124 return fig_passed, fig_duration