generate_report.py

import os
import re
import xml.etree.ElementTree as ET

TEST_DIR = "build/test-results/test"


def generate_report_table(headers, data):
    header_row = "| " + " | ".join(headers) + " |\n"
    separator_row = "| " + " | ".join(["---"] * len(headers)) + " |\n"
    f.write(header_row)
    f.write(separator_row)
    for row in data:
        row_bytes = [
            cell.encode("utf-8").decode("utf-8") if isinstance(cell, str) else str(cell)
            for cell in row
        ]
        f.write("| " + " | ".join(row_bytes) + " |\n")


def pretty_test_name(test_name):
    test_name = re.sub(r"T(\d+)", r"Task\1:", test_name)
    test_name = re.sub(r"(?<=[a-z])(?=[A-Z])", r" ", test_name)
    test_name = test_name.replace("test", "")
    test_name = test_name.replace("Test", "")
    test_name = test_name.replace("()", "")
    return test_name.strip()


def pretty_message(message):
    message_parts = message.split(":")
    error_type = message_parts[0]
    error_message = message_parts[1]
    if "AssertionFailedError" in error_type:
        error_message = error_message.split("==>")[0]
    if "Exception" in error_type:
        error_message = "⚠️ " + error_type.split(".")[-1]
    return error_message


def extract_task_number(filename):
    match = re.search(r"T(\d+)", filename)
    return int(match.group(1)) if match else float("inf")


f = open("report.md", "w", encoding="utf-8")
f.write("\n# Report\n\n")
headers = ["Test", "Status", "Reason"]

filenames = [filename for filename in os.listdir(TEST_DIR) if filename.endswith(".xml")]
filenames.sort(key=extract_task_number)

for filename in filenames:
    data = []
    title = filename.split(".")[-2]
    f.write("## " + pretty_test_name(title) + "\n\n")
    root = ET.parse(os.path.join(TEST_DIR, filename))
    for e in root.findall(".//testcase"):
        failure = e.find("failure")
        if failure is not None:
            test_name = pretty_test_name(e.get("name"))
            message = pretty_message(failure.get("message"))
            data.append([test_name, "❌ Failed", message])
        else:
            test_name = pretty_test_name(e.get("name"))
            message = "Passed"
            data.append([test_name, "✅ Passed", "-"])

    generate_report_table(headers, data)

f.close()