[debuginfo-tests][dexter] Add a test tool --calculate-average option
Differential Revision: https://reviews.llvm.org/D75235
GitOrigin-RevId: 527bd24c3b41a54cef241ea2a1be53f4a4dedb17
diff --git a/dexter/dex/tools/test/Tool.py b/dexter/dex/tools/test/Tool.py
index 9df1d08..91c9dd4 100644
--- a/dexter/dex/tools/test/Tool.py
+++ b/dexter/dex/tools/test/Tool.py
@@ -6,6 +6,7 @@
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
"""Test tool."""
+import math
import os
import csv
import pickle
@@ -94,6 +95,9 @@
help='exit with status FAIL(2) if the test result'
' is less than this value.',
metavar='<float>')
+ parser.add_argument('--calculate-average',
+ action="store_true",
+ help='calculate the average score of every test run')
super(Tool, self).add_tool_arguments(parser, defaults)
def _build_test_case(self):
@@ -226,6 +230,19 @@
if not options.verbose:
self.context.o.auto('\n')
+ if options.calculate_average:
+ # Calculate and print the average score
+ score_sum = 0.0
+ num_tests = 0
+ for test_case in self._test_cases:
+ score = test_case.score
+ if not test_case.error and not math.isnan(score):
+ score_sum += test_case.score
+ num_tests += 1
+
+ if num_tests != 0:
+ print("@avg: ({:.4f})".format(score_sum/num_tests))
+
summary_path = os.path.join(options.results_directory, 'summary.csv')
with open(summary_path, mode='w', newline='') as fp:
writer = csv.writer(fp, delimiter=',')
