Added counts to the summary legend

summary_page.py

@@ -38,15 +38,17 @@ LABEL_OFFSET = 1  # Distance of label from bar
 FAILURE_RATE_FONT_SIZE = 28
 
 
-def calculate_overall_failure_rates(df: pd.DataFrame, available_models: list[str]) -> tuple[float, float]:
+def get_overall_stats(df: pd.DataFrame, available_models: list[str]) -> tuple[list[int], list[int]]:
     """Calculate overall failure rates for AMD and NVIDIA across all models."""
     if df.empty or not available_models:
         return 0.0, 0.0
     
-    total_amd_tests = 0
-    total_amd_failures = 0
-    total_nvidia_tests = 0
-    total_nvidia_failures = 0
+    total_amd_passed = 0
+    total_amd_failed = 0
+    total_amd_skipped = 0
+    total_nvidia_passed = 0
+    total_nvidia_failed = 0
+    total_nvidia_skipped = 0
     
     for model_name in available_models:
         if model_name not in df.index:
@@ -56,21 +58,16 @@ def calculate_overall_failure_rates(df: pd.DataFrame, available_models: list[str
         amd_stats, nvidia_stats = extract_model_data(row)[:2]
         
         # AMD totals
-        amd_total = amd_stats['passed'] + amd_stats['failed'] + amd_stats['error']
-        if amd_total > 0:
-            total_amd_tests += amd_total
-            total_amd_failures += amd_stats['failed'] + amd_stats['error']
-        
+        total_amd_passed += amd_stats['passed']
+        total_amd_failed += amd_stats['failed'] + amd_stats['error']
+        total_amd_skipped += amd_stats['skipped']
+
         # NVIDIA totals
-        nvidia_total = nvidia_stats['passed'] + nvidia_stats['failed'] + nvidia_stats['error']
-        if nvidia_total > 0:
-            total_nvidia_tests += nvidia_total
-            total_nvidia_failures += nvidia_stats['failed'] + nvidia_stats['error']
-    
-    amd_failure_rate = (total_amd_failures / total_amd_tests * 100) if total_amd_tests > 0 else 0.0
-    nvidia_failure_rate = (total_nvidia_failures / total_nvidia_tests * 100) if total_nvidia_tests > 0 else 0.0
+        total_nvidia_passed += nvidia_stats['passed']
+        total_nvidia_failed += nvidia_stats['failed'] + nvidia_stats['error']
+        total_nvidia_skipped += nvidia_stats['skipped']
     
-    return amd_failure_rate, nvidia_failure_rate
+    return [total_amd_passed, total_amd_failed, total_amd_skipped], [total_nvidia_passed, total_nvidia_failed, total_nvidia_skipped]
 
 
 def draw_text_and_bar(
@@ -118,7 +115,12 @@ def create_summary_page(df: pd.DataFrame, available_models: list[str]) -> plt.Fi
         return fig
     
     # Calculate overall failure rates
-    amd_failure_rate, nvidia_failure_rate = calculate_overall_failure_rates(df, available_models)
+    amd_counts, nvidia_counts = get_overall_stats(df, available_models)
+
+    amd_failure_rate = (amd_counts[1] / sum(amd_counts)) if sum(amd_counts) > 0 else 0.0
+    amd_failure_rate *= 100
+    nvidia_failure_rate = (nvidia_counts[1] / sum(nvidia_counts)) if sum(nvidia_counts) > 0 else 0.0
+    nvidia_failure_rate *= 100
 
     # Calculate dimensions for N-column layout
     model_count = len(available_models)
@@ -181,6 +183,26 @@ def create_summary_page(df: pd.DataFrame, available_models: list[str]) -> plt.Fi
         visible_model_count += 1
 
 
+    # Add AMD and NVIDIA test totals in the bottom left
+    # Calculate line spacing to align middle with legend
+    line_height = 0.4  # Height between lines
+    legend_y = max_y + 1
+
+    # Position the two lines so their middle aligns with legend_y
+    amd_y = legend_y - line_height / 2
+    nvidia_y = legend_y + line_height / 2
+
+    amd_totals_text =    f"AMD Tests    - Passed: {amd_counts[0]}, Failed: {amd_counts[1]}, Skipped: {amd_counts[2]}"
+    nvidia_totals_text = f"NVIDIA Tests - Passed: {nvidia_counts[0]}, Failed: {nvidia_counts[1]}, Skipped: {nvidia_counts[2]}"
+
+    ax.text(0, amd_y, amd_totals_text,
+           ha='left', va='bottom', color='#CCCCCC',
+           fontsize=14, fontfamily='monospace')
+
+    ax.text(0, nvidia_y, nvidia_totals_text,
+           ha='left', va='bottom', color='#CCCCCC',
+           fontsize=14, fontfamily='monospace')
+
     # Add legend horizontally in bottom right corner
     patch_height = 0.3
     patch_width = 3
@@ -190,12 +212,6 @@ def create_summary_page(df: pd.DataFrame, available_models: list[str]) -> plt.Fi
     legend_spacing = 10
     legend_font_size = 15
 
-    # Add failure rate explanation text on the left
-    # explanation_text = "Failure rate = failed / (passed + failed)"
-    # ax.text(0, legend_y, explanation_text, 
-    #        ha='left', va='bottom', color='#CCCCCC', 
-    #        fontsize=legend_font_size, fontfamily='monospace', style='italic')
-    
     # Legend entries
     legend_items = [
         ('passed', 'Passed'),