refactor: reorganize project structure by moving core modules and update import paths in API server

2026-07-18 01:24:30 +07:00
parent abab846884
commit a82b2f6fa5
155 changed files with 25 additions and 370 deletions
@@ -0,0 +1,113 @@
+#!/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()