Doctorado-ML/benchmark
1
0
Fork 0
mirror of https://github.com/Doctorado-ML/benchmark.git synced 2025-08-20 09:55:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Continue script testing

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-05-07 02:08:11 +02:00
parent df757fefcd
commit 31059ea117
2 changed files with 92 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

82
benchmark/tests/scripts/Be_Report_test.py
View File

@@ -1,4 +1,6 @@
import os
from openpyxl import load_workbook
from ...Utils import Folders
from ..TestBase import TestBase
@@ -6,6 +8,14 @@ class BeReportTest(TestBase):
def setUp(self):
self.prepare_scripts_env()
def tearDown(self) -> None:
files = [
"results_accuracy_ODTE_Galgo_2022-04-20_10:52:20_0.sql",
"results_accuracy_STree_iMac27_2021-09-30_11:42:07_0.xlsx",
]
self.remove_files(files, Folders.results)
return super().tearDown()
def test_be_report(self):
file_name = os.path.join(
"results",
@@ -18,8 +28,76 @@ class BeReportTest(TestBase):
def test_be_report_compare(self):
file_name = "results_accuracy_STree_iMac27_2021-09-30_11:42:07_0.json"
stdout, stderr = self.execute_script(
"be_report",
["-f", file_name, "-c", "1"],
"be_report", ["-f", file_name, "-c", "1"]
)
self.assertEqual(stderr.getvalue(), "")
self.check_output_file(stdout, "report_compared")
def test_be_report_datatsets(self):
stdout, stderr = self.execute_script("be_report", [])
self.assertEqual(stderr.getvalue(), "")
self.check_output_file(stdout, "report_datasets")
def test_be_report_best(self):
stdout, stderr = self.execute_script(
"be_report", ["-s", "accuracy", "-m", "STree", "-b", "1"]
)
self.assertEqual(stderr.getvalue(), "")
self.check_output_file(stdout, "report_best")
def test_be_report_grid(self):
stdout, stderr = self.execute_script(
"be_report", ["-s", "accuracy", "-m", "STree", "-g", "1"]
)
self.assertEqual(stderr.getvalue(), "")
self.check_output_file(stdout, "report_grid")
def test_be_report_best_both(self):
stdout, stderr = self.execute_script(
"be_report",
["-s", "accuracy", "-m", "STree", "-b", "1", "-g", "1"],
)
self.assertEqual(stderr.getvalue(), "")
self.check_output_file(stdout, "report_best")
def test_be_report_excel_compared(self):
file_name = "results_accuracy_STree_iMac27_2021-09-30_11:42:07_0.json"
stdout, stderr = self.execute_script(
"be_report",
["-f", file_name, "-x", "1", "-c", "1"],
)
file_name = os.path.join(
Folders.results, file_name.replace(".json", ".xlsx")
)
book = load_workbook(file_name)
sheet = book["STree"]
self.check_excel_sheet(sheet, "excel_compared")
self.assertEqual(stderr.getvalue(), "")
self.check_output_file(stdout, "report_compared")
def test_be_report_excel(self):
file_name = "results_accuracy_STree_iMac27_2021-09-30_11:42:07_0.json"
stdout, stderr = self.execute_script(
"be_report",
["-f", file_name, "-x", "1"],
)
file_name = os.path.join(
Folders.results, file_name.replace(".json", ".xlsx")
)
book = load_workbook(file_name)
sheet = book["STree"]
self.check_excel_sheet(sheet, "excel")
self.assertEqual(stderr.getvalue(), "")
self.check_output_file(stdout, "report")
def test_be_report_sql(self):
file_name = "results_accuracy_ODTE_Galgo_2022-04-20_10:52:20_0.json"
stdout, stderr = self.execute_script(
"be_report",
["-f", file_name, "-q", "1"],
)
file_name = os.path.join(
Folders.results, file_name.replace(".json", ".sql")
)
self.check_file_file(file_name, "sql")
self.assertEqual(stderr.getvalue(), "")