Victor Phan
114 lines
3.8 KiB
Python
Executable File
114 lines
3.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Test script to verify training API endpoints
|
|
"""
|
|
|
|
import requests
|
|
import json
|
|
|
|
API_BASE = "http://localhost:8000/api"
|
|
|
|
def test_training_labels():
|
|
"""Test /api/training/labels endpoint"""
|
|
print("=" * 70)
|
|
print("TEST 1: Getting training labels")
|
|
print("=" * 70)
|
|
|
|
response = requests.get(f"{API_BASE}/training/labels")
|
|
if response.ok:
|
|
data = response.json()
|
|
print(f"✅ Success! Found {data['count']} labels:")
|
|
for label in data['labels']:
|
|
print(f" {label['code']}: {label['name']}")
|
|
else:
|
|
print(f"❌ Error: {response.status_code}")
|
|
print()
|
|
|
|
def test_training_files():
|
|
"""Test /api/training/files endpoint"""
|
|
print("=" * 70)
|
|
print("TEST 2: Getting training files")
|
|
print("=" * 70)
|
|
|
|
response = requests.get(f"{API_BASE}/training/files")
|
|
if response.ok:
|
|
data = response.json()
|
|
print(f"✅ Success! Found {data['count']} training files:")
|
|
for file in data['files']:
|
|
print(f"\n 📄 {file['filename']}")
|
|
print(f" Size: {file['size_mb']} MB")
|
|
if 'point_count' in file:
|
|
print(f" Points: {file['point_count']}")
|
|
print(f" Label column: {file.get('label_column', 'N/A')}")
|
|
print(f" Unique labels: {file.get('label_count', 0)}")
|
|
else:
|
|
print(f"❌ Error: {response.status_code}")
|
|
print()
|
|
|
|
def test_shapefile_labels(filename="ST_training_data_updated_1130points_new.shp"):
|
|
"""Test /api/training/shapefile/{filename}/labels endpoint"""
|
|
print("=" * 70)
|
|
print(f"TEST 3: Getting labels from shapefile: {filename}")
|
|
print("=" * 70)
|
|
|
|
response = requests.get(f"{API_BASE}/training/shapefile/{filename}/labels")
|
|
if response.ok:
|
|
data = response.json()
|
|
print(f"✅ Success!")
|
|
print(f" Filename: {data['filename']}")
|
|
print(f" Points: {data['point_count']}")
|
|
print(f" Label column: {data['label_column']}")
|
|
print(f" Unique labels: {data['label_count']}")
|
|
print(f" Bbox: {data['bbox']}")
|
|
print(f"\n Labels distribution:")
|
|
for label in data['labels']:
|
|
mapped = "✅" if label['mapped'] else "⚠️"
|
|
print(f" {mapped} {label['name']}: {label['count']} points (code: {label['code']})")
|
|
else:
|
|
print(f"❌ Error: {response.status_code}")
|
|
print(response.text)
|
|
print()
|
|
|
|
def test_config_presets():
|
|
"""Test /api/config/presets endpoint"""
|
|
print("=" * 70)
|
|
print("TEST 4: Getting config presets")
|
|
print("=" * 70)
|
|
|
|
response = requests.get(f"{API_BASE}/config/presets")
|
|
if response.ok:
|
|
data = response.json()
|
|
print(f"✅ Success! Found {len(data['presets'])} presets:")
|
|
for preset in data['presets']:
|
|
print(f"\n 📋 {preset['name']}")
|
|
config = preset['config']
|
|
print(f" Bbox: [{config['min_lon']}, {config['min_lat']}, {config['max_lon']}, {config['max_lat']}]")
|
|
print(f" Time: {config['start_date']} → {config['end_date']}")
|
|
print(f" Resolution: {config['resolution']}m")
|
|
else:
|
|
print(f"❌ Error: {response.status_code}")
|
|
print()
|
|
|
|
if __name__ == "__main__":
|
|
print("\n" + "=" * 70)
|
|
print("🧪 TESTING TRAINING API ENDPOINTS")
|
|
print("=" * 70 + "\n")
|
|
|
|
try:
|
|
test_training_labels()
|
|
test_training_files()
|
|
test_shapefile_labels()
|
|
test_config_presets()
|
|
|
|
print("=" * 70)
|
|
print("✅ ALL TESTS COMPLETED!")
|
|
print("=" * 70)
|
|
|
|
except requests.exceptions.ConnectionError:
|
|
print("\n❌ Error: Cannot connect to API server")
|
|
print("Make sure the server is running: python api_server.py")
|
|
except Exception as e:
|
|
print(f"\n❌ Error: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|