test_recording.py

#!/usr/bin/env python3
"""
Test script to verify recording functionality
"""

import os
import sys
import pygame
import time

sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from demo import DemoRecorder
from main import *

def test_recording():
    """Test the recording functionality"""
    print("🧪 Testing recording functionality...")
    
    # Initialize pygame
    pygame.init()
    
    # Create a small demo
    draw_info = DrawInformation(800, 600, [5, 2, 8, 1, 9])
    recorder = DemoRecorder()
    
    # Test recording
    recorder.start_recording("Test Algorithm")
    
    # Capture a few frames
    for i in range(5):
        draw_complete(draw_info, "Test Algorithm", True, 5)
        recorder.capture_frame(draw_info.window)
        time.sleep(0.1)
    
    recorder.stop_recording()
    
    # Check if files were created
    test_dir = "recordings/test_algorithm"
    if os.path.exists(test_dir):
        files = os.listdir(test_dir)
        png_files = [f for f in files if f.endswith('.png')]
        print(f"✅ Recording test passed! Created {len(png_files)} frames")
        
        # Test GIF creation
        try:
            recorder.create_gif("Test Algorithm")
            gif_path = "recordings/test_algorithm.gif"
            if os.path.exists(gif_path):
                print("✅ GIF creation test passed!")
            else:
                print("❌ GIF creation test failed!")
        except Exception as e:
            print(f"⚠️ GIF creation test failed: {e}")
            print("   (This is normal if Pillow is not installed)")
    else:
        print("❌ Recording test failed! No frames created")
    
    pygame.quit()
    print("🧪 Recording test completed!")

if __name__ == "__main__":
    test_recording()