62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
|
import requests
|
||
|
|
import json
|
||
|
|
import time
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
|
||
|
|
# Start API in background
|
||
|
|
print("Starting API...")
|
||
|
|
api_process = subprocess.Popen([sys.executable, "src/inference/app.py"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||
|
|
|
||
|
|
# Wait for startup
|
||
|
|
time.sleep(5)
|
||
|
|
|
||
|
|
url = "http://localhost:5000/predict"
|
||
|
|
|
||
|
|
# Test Case 1: CT Advantage (3v1, high health)
|
||
|
|
payload_ct_win = {
|
||
|
|
"game_time": 60.0,
|
||
|
|
"players": [
|
||
|
|
{"team_num": 3, "is_alive": True, "health": 100},
|
||
|
|
{"team_num": 3, "is_alive": True, "health": 100},
|
||
|
|
{"team_num": 3, "is_alive": True, "health": 90},
|
||
|
|
{"team_num": 2, "is_alive": True, "health": 50}
|
||
|
|
]
|
||
|
|
}
|
||
|
|
|
||
|
|
# Test Case 2: T Advantage (1v3)
|
||
|
|
payload_t_win = {
|
||
|
|
"game_time": 45.0,
|
||
|
|
"players": [
|
||
|
|
{"team_num": 3, "is_alive": True, "health": 10},
|
||
|
|
{"team_num": 2, "is_alive": True, "health": 100},
|
||
|
|
{"team_num": 2, "is_alive": True, "health": 100},
|
||
|
|
{"team_num": 2, "is_alive": True, "health": 100}
|
||
|
|
]
|
||
|
|
}
|
||
|
|
|
||
|
|
def test_payload(name, payload):
|
||
|
|
print(f"\n--- Testing {name} ---")
|
||
|
|
try:
|
||
|
|
response = requests.post(url, json=payload, timeout=2)
|
||
|
|
print("Status Code:", response.status_code)
|
||
|
|
if response.status_code == 200:
|
||
|
|
print("Response:", json.dumps(response.json(), indent=2))
|
||
|
|
else:
|
||
|
|
print("Error:", response.text)
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Request failed: {e}")
|
||
|
|
|
||
|
|
try:
|
||
|
|
test_payload("CT Advantage Scenario", payload_ct_win)
|
||
|
|
test_payload("T Advantage Scenario", payload_t_win)
|
||
|
|
finally:
|
||
|
|
print("\nStopping API...")
|
||
|
|
api_process.terminate()
|
||
|
|
try:
|
||
|
|
outs, errs = api_process.communicate(timeout=2)
|
||
|
|
print("API Output:", outs.decode())
|
||
|
|
print("API Errors:", errs.decode())
|
||
|
|
except:
|
||
|
|
api_process.kill()
|