Files
LM-Gambit/results_chart.py
T
2025-10-03 04:28:51 -04:00

47 lines
1.7 KiB
Python

import tkinter as tk
from tkinter import ttk
from pathlib import Path
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
import re
def parse_report(report_path: Path):
# Simple parser for markdown report: looks for lines like 'Pass: 8/10', 'Fail: 2/10', etc.
pass_count = 0
fail_count = 0
total = 0
with open(report_path, 'r', encoding='utf-8') as f:
for line in f:
m = re.match(r'Pass: (\d+)/(\d+)', line)
if m:
pass_count = int(m.group(1))
total = int(m.group(2))
m = re.match(r'Fail: (\d+)/(\d+)', line)
if m:
fail_count = int(m.group(1))
return pass_count, fail_count, total
class ResultsChartDialog(tk.Toplevel):
def __init__(self, parent, report_path: Path):
super().__init__(parent)
self.title(f"Test Results Chart: {report_path.name}")
self.geometry("600x500")
self.transient(parent)
self.grab_set()
pass_count, fail_count, total = parse_report(report_path)
fig, ax = plt.subplots(figsize=(5,4))
ax.bar(['Pass', 'Fail'], [pass_count, fail_count], color=['#4caf50', '#f44336'])
ax.set_ylabel('Count')
ax.set_title(f"Test Results ({pass_count+fail_count}/{total} completed)")
for i, v in enumerate([pass_count, fail_count]):
ax.text(i, v + 0.1, str(v), ha='center', va='bottom', fontweight='bold')
fig.tight_layout()
canvas = FigureCanvasTkAgg(fig, master=self)
canvas.draw()
canvas.get_tk_widget().pack(fill="both", expand=True)
ttk.Button(self, text="Close", command=self.destroy).pack(pady=8)