scottk/wlan-lanforge-scripts
1
0
Fork 0
mirror of https://github.com/Telecominfraproject/wlan-lanforge-scripts.git synced 2025-11-01 11:18:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

x-axis values of lf_bar_graph added to csv generation

Browse Source
This commit is contained in:
anil-tegala
2021-08-06 01:40:14 +05:30
parent 184c49daf7
commit 4b0eb41019
2 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
py-scripts/lf_graph.py
View File

@@ -35,8 +35,8 @@ class lf_bar_graph():
def __init__(self, _data_set=[[30.4, 55.3, 69.2, 37.1], [45.1, 67.2, 34.3, 22.4], [22.5, 45.6, 12.7, 34.8]],
_xaxis_name="x-axis",
_yaxis_name="y-axis",
_xaxis_categories=[1, 2, 3, 4, 5],
_xaxis_label=["a", "b", "c", "d", "e"],
_xaxis_categories=[1, 2, 3, 4],
_xaxis_label=["a", "b", "c", "d"],
_graph_title="",
_title_size=16,
_graph_image_name="image_name",
@@ -48,7 +48,7 @@ class lf_bar_graph():
_color_name=['lightcoral', 'darkgrey', 'r', 'g', 'b', 'y'],
_figsize=(10, 5),
_show_bar_value=False,
_xaxis_step=5,
_xaxis_step=1,
_xticks_font = None,
_text_font=None,
_text_rotation=None,
@@ -139,11 +139,18 @@ class lf_bar_graph():
plt.close()
print("{}.png".format(self.graph_image_name))
if self.enable_csv:
if self.data_set is not None:
self.lf_csv.columns = self.label
self.lf_csv.rows = self.data_set
self.lf_csv.filename = f"{self.graph_image_name}.csv"
self.lf_csv.generate_csv()
if self.data_set is not None and self.xaxis_categories is not None:
if len(self.xaxis_categories) == len(self.data_set[0]):
self.lf_csv.columns = []
self.lf_csv.rows = []
self.lf_csv.columns.append(self.xaxis_name)
self.lf_csv.columns.extend(self.label)
self.lf_csv.rows.append(self.xaxis_categories)
self.lf_csv.rows.extend(self.data_set)
self.lf_csv.filename = f"{self.graph_image_name}.csv"
self.lf_csv.generate_csv()
else:
raise ValueError("Length and x-axis values and y-axis values should be same.")
else:
print("No Dataset Found")
print("{}.csv".format(self.graph_image_name))
@@ -364,7 +371,8 @@ if __name__ == "__main__":
_graph_image_name="Bi-single_radio_2.4GHz",
_label=["bi-downlink", "bi-uplink", 'uplink'],
_color=None,
_color_edge='red')
_color_edge='red',
_enable_csv=True)
graph_html_obj = """
<img align='center' style='padding:15;margin:5;width:1000px;' src=""" + "%s" % (graph.build_bar_graph()) + """ border='1' />
<br><br>

7
py-scripts/lf_report_test.py
View File

@@ -107,7 +107,8 @@ if __name__ == "__main__":
_legend_loc="best",
_legend_box=(1,1),
_legend_ncol=1,
_legend_fontsize=None)
_legend_fontsize=None,
_enable_csv=True)
graph_png = graph.build_bar_graph()
@@ -116,7 +117,9 @@ if __name__ == "__main__":
report.set_graph_image(graph_png)
# need to move the graph image to the results
report.move_graph_image()
if graph.enable_csv:
report.set_csv_filename(graph_png)
report.move_csv_file()
report.build_graph()
set1 = [1, 2, 3, 4]
set2 = [[45, 67, 45, 34], [34, 56, 45, 34], [45, 78, 23, 45]]