xref: /aosp_15_r20/external/pytorch/tools/code_coverage/oss_coverage.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1#!/usr/bin/env python3
2import time
3
4from package.oss.cov_json import get_json_report  # type: ignore[import]
5from package.oss.init import initialization  # type: ignore[import]
6from package.tool.summarize_jsons import summarize_jsons  # type: ignore[import]
7from package.util.setting import TestPlatform  # type: ignore[import]
8from package.util.utils import print_time  # type: ignore[import]
9
10
11def report_coverage() -> None:
12    start_time = time.time()
13    (options, test_list, interested_folders) = initialization()
14    # run cpp tests
15    get_json_report(test_list, options)
16    # collect coverage data from json profiles
17    if options.need_summary:
18        summarize_jsons(test_list, interested_folders, [""], TestPlatform.OSS)
19    # print program running time
20    print_time("Program Total Time: ", start_time)
21
22
23if __name__ == "__main__":
24    report_coverage()
25