mirror of
https://github.com/bentoml/OpenLLM.git
synced 2026-01-20 05:18:50 -05:00
72 lines
2.3 KiB
Python
Executable File
72 lines
2.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright 2023 BentoML Team. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from __future__ import annotations
|
|
|
|
from decimal import ROUND_DOWN
|
|
from decimal import Decimal
|
|
from pathlib import Path
|
|
|
|
import orjson
|
|
|
|
|
|
PRECISION = Decimal(".01")
|
|
|
|
ROOT = Path(__file__).resolve().parent.parent
|
|
|
|
|
|
def main():
|
|
coverage_summary = ROOT / "coverage-summary.json"
|
|
|
|
coverage_data = orjson.loads(coverage_summary.read_text(encoding="utf-8"))
|
|
total_data = coverage_data.pop("total")
|
|
|
|
lines = [
|
|
"\n",
|
|
"Package | Statements\n",
|
|
"------- | ----------\n",
|
|
]
|
|
|
|
for package, data in sorted(coverage_data.items()):
|
|
statements_covered = data["statements_covered"]
|
|
statements = data["statements"]
|
|
|
|
rate = Decimal(statements_covered) / Decimal(statements) * 100
|
|
rate = rate.quantize(PRECISION, rounding=ROUND_DOWN)
|
|
lines.append(
|
|
f"{package} | {100 if rate == 100 else rate}% ({statements_covered} / {statements})\n" # noqa: PLR2004
|
|
)
|
|
|
|
total_statements_covered = total_data["statements_covered"]
|
|
total_statements = total_data["statements"]
|
|
total_rate = Decimal(total_statements_covered) / Decimal(total_statements) * 100
|
|
total_rate = total_rate.quantize(PRECISION, rounding=ROUND_DOWN)
|
|
color = "ok" if float(total_rate) >= 95 else "critical"
|
|
lines.insert(0, f"\n")
|
|
|
|
lines.append(
|
|
f"**Summary** | {100 if total_rate == 100 else total_rate}% "
|
|
f"({total_statements_covered} / {total_statements})\n"
|
|
)
|
|
|
|
coverage_report = ROOT / "coverage-report.md"
|
|
with coverage_report.open("w", encoding="utf-8") as f:
|
|
f.write("".join(lines))
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|