File size: 1,198 Bytes
2532605 | 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 | import requests
import json
# Test 1: Health check
print('=' * 70)
print('TEST 1: Health Check')
print('=' * 70)
resp = requests.get('http://127.0.0.1:8000/health')
print(json.dumps(resp.json(), indent=2))
print()
# Test 2: List drivers
print('=' * 70)
print('TEST 2: List Drivers (2023)')
print('=' * 70)
resp = requests.get('http://127.0.0.1:8000/drivers?season=2023')
drivers = resp.json()
print('Found {} drivers'.format(len(drivers)))
print('First 3:')
for d in drivers[:3]:
print(' {}'.format(d))
print()
# Test 3: List races
print('=' * 70)
print('TEST 3: List Races (2023)')
print('=' * 70)
resp = requests.get('http://127.0.0.1:8000/races/2023')
races = resp.json()
print('Found {} races'.format(len(races)))
print('First 3:')
for r in races[:3]:
print(' Round {}: {}'.format(r.get('round'), r.get('name')))
print()
# Test 4: MAIN TEST
print('=' * 70)
print('TEST 4: Natural Language Prediction')
print('=' * 70)
query = 'What is Max Verstappens win probability at Monaco 2023?'
print('Query: {}'.format(query))
resp = requests.post(
'http://127.0.0.1:8000/predict/f1',
json={'query': query},
timeout=30
)
result = resp.json()
print(json.dumps(result, indent=2))
|