From e45ef1c9faa2205ca6d950304e2deed77f577ebf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= Date: Mon, 9 May 2022 12:02:33 +0200 Subject: [PATCH] Add file not found manage to be_report --- benchmark/scripts/be_report.py | 30 +++++++++++++---------- benchmark/tests/scripts/Be_Report_test.py | 5 ++++ 2 files changed, 22 insertions(+), 13 deletions(-) diff --git a/benchmark/scripts/be_report.py b/benchmark/scripts/be_report.py index d62d9ad..6abdf40 100755 --- a/benchmark/scripts/be_report.py +++ b/benchmark/scripts/be_report.py @@ -27,16 +27,20 @@ def main(args_test=None): report = ReportBest(args.score, args.model, args.best, args.grid) report.report() else: - report = Report(args.file, args.compare) - report.report() - if args.excel: - excel = Excel( - file_name=args.file, - compare=args.compare, - ) - excel.report() - is_test = args_test is not None - Files.open(excel.get_file_name(), is_test) - if args.sql: - sql = SQL(args.file) - sql.report() + try: + report = Report(args.file, args.compare) + except FileNotFoundError as e: + print(e) + else: + report.report() + if args.excel: + excel = Excel( + file_name=args.file, + compare=args.compare, + ) + excel.report() + is_test = args_test is not None + Files.open(excel.get_file_name(), is_test) + if args.sql: + sql = SQL(args.file) + sql.report() diff --git a/benchmark/tests/scripts/Be_Report_test.py b/benchmark/tests/scripts/Be_Report_test.py index 5308fa0..5fe2680 100644 --- a/benchmark/tests/scripts/Be_Report_test.py +++ b/benchmark/tests/scripts/Be_Report_test.py @@ -25,6 +25,11 @@ class BeReportTest(TestBase): self.assertEqual(stderr.getvalue(), "") self.check_output_file(stdout, "report") + def test_be_report_not_found(self): + stdout, stderr = self.execute_script("be_report", ["-f", "unknown"]) + self.assertEqual(stderr.getvalue(), "") + self.assertEqual(stdout.getvalue(), "unknown does not exists!\n") + 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(