Spaces:
Sleeping
Sleeping
File size: 5,187 Bytes
120c870 |
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 |
"""
Test script for SuperKart Backend API
Simple test script to verify the API endpoints are working correctly.
Run this after starting the Flask application.
"""
import requests
from typing import Any, Dict, List, Tuple
# API base URL
BASE_URL = "http://localhost:7860"
def test_health_check() -> bool:
"""Test the health check endpoint."""
print("🔍 Testing health check endpoint...")
try:
response = requests.get(f"{BASE_URL}/")
print(f"Status: {response.status_code}")
print(f"Response: {response.json()}")
return response.status_code == 200
except Exception as e:
print(f"❌ Health check failed: {e}")
return False
def test_features_endpoint() -> bool:
"""Test the features information endpoint."""
print("\n🔍 Testing features endpoint...")
try:
response = requests.get(f"{BASE_URL}/features")
print(f"Status: {response.status_code}")
data = response.json()
print(f"Required features: {len(data['required_features'])}")
return response.status_code == 200
except Exception as e:
print(f"❌ Features endpoint failed: {e}")
return False
def test_single_prediction() -> bool:
"""Test single prediction endpoint."""
print("\n🔍 Testing single prediction endpoint...")
# Example input data
test_data: Dict[str, Any] = {
"Product_Weight": 12.66,
"Product_Sugar_Content": "Low Sugar",
"Product_Allocated_Area": 0.027,
"Product_Type": "Frozen Foods",
"Product_MRP": 117.08,
"Store_Establishment_Year": 2009,
"Store_Size": "Medium",
"Store_Location_City_Type": "Tier 2",
"Store_Type": "Supermarket Type2",
}
try:
response = requests.post(f"{BASE_URL}/predict", json=test_data)
print(f"Status: {response.status_code}")
if response.status_code == 200:
result = response.json()
print(f"Predicted sales: ${result['predicted_sales']:.2f}")
return True
else:
print(f"Error: {response.json()}")
return False
except Exception as e:
print(f"❌ Single prediction failed: {e}")
return False
def test_batch_prediction() -> bool:
"""Test batch prediction endpoint."""
print("\n🔍 Testing batch prediction endpoint...")
# Example batch data
batch_data: Dict[str, List[Dict[str, Any]]] = {
"predictions": [
{
"Product_Weight": 12.66,
"Product_Sugar_Content": "Low Sugar",
"Product_Allocated_Area": 0.027,
"Product_Type": "Frozen Foods",
"Product_MRP": 117.08,
"Store_Establishment_Year": 2009,
"Store_Size": "Medium",
"Store_Location_City_Type": "Tier 2",
"Store_Type": "Supermarket Type2",
},
{
"Product_Weight": 16.54,
"Product_Sugar_Content": "Low Sugar",
"Product_Allocated_Area": 0.144,
"Product_Type": "Dairy",
"Product_MRP": 171.43,
"Store_Establishment_Year": 1999,
"Store_Size": "Medium",
"Store_Location_City_Type": "Tier 1",
"Store_Type": "Departmental Store",
},
],
}
try:
response = requests.post(f"{BASE_URL}/predict/batch", json=batch_data)
print(f"Status: {response.status_code}")
if response.status_code == 200:
result = response.json()
print(f"Successful predictions: {result['successful_predictions']}")
print(f"Failed predictions: {result['failed_predictions']}")
# Print each prediction value
if "results" in result:
for pred in result["results"]:
idx = pred.get("index", "?")
val = pred.get("predicted_sales", "?")
print(f"Prediction {idx}: ${val:.2f}")
return True
else:
print(f"Error: {response.json()}")
return False
except Exception as e:
print(f"❌ Batch prediction failed: {e}")
return False
def run_all_tests() -> None:
"""Run all API tests."""
print("🚀 Starting SuperKart API Tests\n")
tests: List[Tuple[str, Any]] = [
("Health Check", test_health_check),
("Features Endpoint", test_features_endpoint),
("Single Prediction", test_single_prediction),
("Batch Prediction", test_batch_prediction),
]
results: List[Tuple[str, bool]] = []
for test_name, test_func in tests:
result = test_func()
results.append((test_name, result))
print("✅ PASSED" if result else "❌ FAILED")
print(f"\n📊 Test Summary:")
passed = sum(1 for _, result in results if result)
total = len(results)
print(f"Passed: {passed}/{total}")
if passed == total:
print("🎉 All tests passed!")
else:
print("⚠️ Some tests failed. Check the API server.")
if __name__ == "__main__":
run_all_tests()
|