speed_test.py

#!/usr/bin/env python3
"""
Speed comparison test for different demo modes
"""

import time
import sys
import os
import pygame

sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from main import (bubble_sort, insertion_sort, quick_sort, generate_list, 
                  DrawInformation)

def time_sorting_algorithm(algorithm, draw_info, name):
    """Time how long it takes to complete a sorting algorithm"""
    start_time = time.time()
    
    # Run the algorithm without visualization
    sorting_generator = algorithm(draw_info, True)
    
    try:
        while True:
            next(sorting_generator)
    except StopIteration:
        pass
    
    end_time = time.time()
    return end_time - start_time

def speed_test():
    """Test the speed of different array sizes"""
    pygame.init()
    
    print("🚀 Speed Test - Sorting Algorithm Performance")
    print("=" * 50)
    
    # Test different array sizes
    test_sizes = [15, 25, 40, 60]
    algorithms = [
        (bubble_sort, "Bubble Sort"),
        (insertion_sort, "Insertion Sort"),
        (quick_sort, "Quick Sort")
    ]
    
    for size in test_sizes:
        print(f"\n📊 Array Size: {size}")
        print("-" * 30)
        
        # Generate test array
        lst = generate_list(size, 1, 100)
        draw_info = DrawInformation(800, 600, lst)
        
        for algorithm, name in algorithms:
            # Reset the array for each algorithm
            draw_info.set_list(lst[:])  # Copy the list
            
            # Time the algorithm
            duration = time_sorting_algorithm(algorithm, draw_info, name)
            
            # Calculate operations
            comparisons = draw_info.stats.comparisons
            swaps = draw_info.stats.swaps
            
            print(f"{name:15} | {duration:6.3f}s | {comparisons:4d} comp | {swaps:4d} swaps")
    
    print(f"\n💡 For fastest demo, use: python demo.py --express")
    print(f"💡 For normal fast demo, use: python demo.py")
    print(f"💡 For detailed demo, increase array size in demo.py")
    
    pygame.quit()

if __name__ == "__main__":
    speed_test()