test_simulation.py

#!/usr/bin/env python3
"""
Test script for the traffic simulation.
This script runs a quick test to verify all components work correctly.
"""

import pygame
import sys
import time
from main import TrafficSimulation

def test_simulation():
    """Test the traffic simulation components."""
    print("Testing Traffic Simulation Components...")
    
    try:
        # Initialize pygame
        pygame.init()
        
        # Create simulation instance
        print("✓ Creating simulation instance...")
        simulation = TrafficSimulation()
        
        # Test component initialization
        print("✓ Testing component initialization...")
        assert simulation.scheduler is not None
        assert simulation.traffic_light is not None
        assert simulation.sync_manager is not None
        assert simulation.dashboard is not None
        assert len(simulation.lanes) == 4
        
        # Test vehicle spawning
        print("✓ Testing vehicle spawning...")
        initial_vehicle_count = len(simulation.vehicles)
        simulation.spawn_vehicle()
        assert len(simulation.vehicles) > initial_vehicle_count
        
        # Test scheduling algorithms
        print("✓ Testing scheduling algorithms...")
        algorithms = ["FCFS", "Round Robin", "Priority", "Dynamic"]
        for algo in algorithms:
            simulation.scheduler.set_algorithm(algo)
            assert simulation.scheduler.get_current_algorithm() == algo
        
        # Test traffic light states
        print("✓ Testing traffic light states...")
        for direction in simulation.lanes.keys():
            state = simulation.traffic_light.get_light_state(direction)
            assert state in ['red', 'yellow', 'green']
        
        # Test synchronization
        print("✓ Testing synchronization...")
        stats = simulation.sync_manager.get_statistics()
        assert 'lock_contention_count' in stats
        assert 'deadlock_resolution_count' in stats
        
        # Test dashboard update
        print("✓ Testing dashboard update...")
        simulation.dashboard.update(
            "FCFS",
            {'north': 2, 'south': 1, 'east': 3, 'west': 0},
            10,
            "Medium"
        )
        
        # Test simulation update (one frame)
        print("✓ Testing simulation update...")
        simulation.update(16)  # ~60 FPS
        
        # Test event handling
        print("✓ Testing event handling...")
        # Simulate key press events
        test_events = [
            pygame.event.Event(pygame.KEYDOWN, key=pygame.K_1),
            pygame.event.Event(pygame.KEYDOWN, key=pygame.K_SPACE),
        ]
        pygame.event.post(test_events[0])
        pygame.event.post(test_events[1])
        simulation.handle_events()
        
        print("✓ All tests passed!")
        print("\n🎉 Traffic Simulation is ready to run!")
        print("Run 'python main.py' to start the simulation.")
        
        return True
        
    except Exception as e:
        print(f"❌ Test failed: {e}")
        import traceback
        traceback.print_exc()
        return False
    
    finally:
        pygame.quit()

if __name__ == "__main__":
    success = test_simulation()
    sys.exit(0 if success else 1)