File size: 6,823 Bytes
a8ba5ce |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 |
"""
Simple Test Script - Verify Metro Scheduling System
Tests core functionality without requiring full API setup
"""
import sys
import traceback
def test_imports():
"""Test that all modules can be imported"""
print("Testing imports...")
try:
from DataService import metro_models
from DataService import metro_data_generator
from DataService import schedule_optimizer
print(" ✓ DataService modules imported successfully")
return True
except Exception as e:
print(f" ✗ Import failed: {e}")
traceback.print_exc()
return False
def test_data_generation():
"""Test data generation"""
print("\nTesting data generation...")
try:
from DataService.metro_data_generator import MetroDataGenerator
generator = MetroDataGenerator(num_trains=10, num_stations=10)
print(f" ✓ Generator created for {len(generator.trainset_ids)} trains")
# Test route generation
route = generator.generate_route()
print(f" ✓ Route generated: {route.name} with {len(route.stations)} stations")
# Test train health
health = generator.generate_train_health_statuses()
print(f" ✓ Generated health status for {len(health)} trains")
# Test certificates
certs = generator.generate_fitness_certificates("TS-001")
print(f" ✓ Generated fitness certificates")
return True
except Exception as e:
print(f" ✗ Data generation failed: {e}")
traceback.print_exc()
return False
def test_schedule_optimization():
"""Test schedule optimization"""
print("\nTesting schedule optimization...")
try:
from DataService.metro_data_generator import MetroDataGenerator
from DataService.schedule_optimizer import MetroScheduleOptimizer
from datetime import datetime
# Setup
generator = MetroDataGenerator(num_trains=15, num_stations=15)
route = generator.generate_route()
health = generator.generate_train_health_statuses()
# Create optimizer
optimizer = MetroScheduleOptimizer(
date=datetime.now().strftime("%Y-%m-%d"),
num_trains=15,
route=route,
train_health=health
)
print(f" ✓ Optimizer created")
# Generate schedule
schedule = optimizer.optimize_schedule(min_service_trains=10, min_standby=2)
print(f" ✓ Schedule generated: {schedule.schedule_id}")
print(f" - Trains in service: {schedule.fleet_summary.revenue_service}")
print(f" - Total planned km: {schedule.optimization_metrics.total_planned_km}")
print(f" - Optimization time: {schedule.optimization_metrics.optimization_runtime_ms} ms")
return True
except Exception as e:
print(f" ✗ Schedule optimization failed: {e}")
traceback.print_exc()
return False
def test_models():
"""Test Pydantic models"""
print("\nTesting data models...")
try:
from DataService.metro_models import (
ScheduleRequest, TrainHealthStatus, Route, Station
)
# Test ScheduleRequest
request = ScheduleRequest(
date="2025-10-25",
num_trains=25,
num_stations=25
)
print(f" ✓ ScheduleRequest model validated")
# Test Station
station = Station(
station_id="STN-001",
name="Test Station",
sequence=1,
distance_from_origin_km=0.0
)
print(f" ✓ Station model validated")
return True
except Exception as e:
print(f" ✗ Model validation failed: {e}")
traceback.print_exc()
return False
def test_json_export():
"""Test JSON export"""
print("\nTesting JSON export...")
try:
import json
from DataService.metro_data_generator import MetroDataGenerator
from DataService.schedule_optimizer import MetroScheduleOptimizer
from datetime import datetime
generator = MetroDataGenerator(num_trains=10, num_stations=10)
route = generator.generate_route()
health = generator.generate_train_health_statuses()
optimizer = MetroScheduleOptimizer(
date=datetime.now().strftime("%Y-%m-%d"),
num_trains=10,
route=route,
train_health=health
)
schedule = optimizer.optimize_schedule()
# Convert to dict and save
schedule_dict = schedule.model_dump()
# Try to serialize to JSON
json_str = json.dumps(schedule_dict, indent=2, default=str)
print(f" ✓ Schedule exported to JSON ({len(json_str)} chars)")
print(f" - Contains {len(schedule_dict['trainsets'])} trainsets")
return True
except Exception as e:
print(f" ✗ JSON export failed: {e}")
traceback.print_exc()
return False
def main():
"""Run all tests"""
print("=" * 70)
print(" METRO SCHEDULING SYSTEM - VERIFICATION TESTS")
print("=" * 70)
tests = [
("Imports", test_imports),
("Data Generation", test_data_generation),
("Schedule Optimization", test_schedule_optimization),
("Data Models", test_models),
("JSON Export", test_json_export)
]
results = []
for name, test_func in tests:
try:
result = test_func()
results.append((name, result))
except Exception as e:
print(f"\n✗ {name} crashed: {e}")
results.append((name, False))
# Summary
print("\n" + "=" * 70)
print(" TEST SUMMARY")
print("=" * 70)
passed = sum(1 for _, result in results if result)
total = len(results)
for name, result in results:
status = "✓ PASS" if result else "✗ FAIL"
print(f" {status}: {name}")
print("\n" + "-" * 70)
print(f" Results: {passed}/{total} tests passed")
if passed == total:
print("\n 🎉 All tests passed! System is ready to use.")
print("\n Next steps:")
print(" 1. Run: python demo_schedule.py")
print(" 2. Run: python run_api.py")
print(" 3. Visit: http://localhost:8000/docs")
else:
print("\n ⚠️ Some tests failed. Please check the errors above.")
print(" Make sure all dependencies are installed:")
print(" pip install -r requirements.txt")
print("=" * 70)
return 0 if passed == total else 1
if __name__ == "__main__":
sys.exit(main())
|