feat: Initial commit of Clutch-IQ project
This commit is contained in:
44
tests/test_spatial_inference.py
Normal file
44
tests/test_spatial_inference.py
Normal file
@@ -0,0 +1,44 @@
|
||||
import requests
|
||||
import json
|
||||
import time
|
||||
|
||||
url = "http://localhost:5000/predict"
|
||||
|
||||
# Scenario: 2v2 Clutch
|
||||
# T Team: Together (Planting B site?)
|
||||
# CT Team: Separated (Retaking?)
|
||||
|
||||
payload_spatial = {
|
||||
"game_time": 90.0,
|
||||
"players": [
|
||||
# T Team (Team 2) - Clumped together
|
||||
{"team_num": 2, "is_alive": True, "health": 100, "X": -1000, "Y": 2000, "Z": 0},
|
||||
{"team_num": 2, "is_alive": True, "health": 100, "X": -1050, "Y": 2050, "Z": 0},
|
||||
|
||||
# CT Team (Team 3) - Far apart (Retaking from different angles)
|
||||
{"team_num": 3, "is_alive": True, "health": 100, "X": 0, "Y": 0, "Z": 0}, # Mid
|
||||
{"team_num": 3, "is_alive": True, "health": 100, "X": -2000, "Y": 3000, "Z": 0} # Flanking
|
||||
]
|
||||
}
|
||||
|
||||
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:
|
||||
data = response.json()
|
||||
print("Response Prediction:", data['prediction'])
|
||||
print("Win Probability:", json.dumps(data['win_probability'], indent=2))
|
||||
print("Spatial Features Calculated:")
|
||||
feats = data['features_used']
|
||||
print(f" Team Distance: {feats.get('team_distance', 'N/A'):.2f}")
|
||||
print(f" T Spread: {feats.get('t_spread', 'N/A'):.2f}")
|
||||
print(f" CT Spread: {feats.get('ct_spread', 'N/A'):.2f}")
|
||||
else:
|
||||
print("Error:", response.text)
|
||||
except Exception as e:
|
||||
print(f"Request failed: {e}")
|
||||
|
||||
if __name__ == "__main__":
|
||||
test_payload("Spatial 2v2 Scenario", payload_spatial)
|
||||
Reference in New Issue
Block a user