Spaces:
Running
Running
File size: 712 Bytes
c31d1ca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from fastapi.testclient import TestClient
from src.main import app # adjust import if needed
client = TestClient(app)
def test_recommend_random():
"""Test that /recommend/random runs and returns a response"""
response = client.get("/recommend/random?genre=rock&n=5")
assert response.status_code == 200
assert "recommendations" in response.json()
def test_recommend_filtered():
"""Test that /recommend/filtered runs with tempo + exercise_id"""
payload = {
"tempo": 120,
"exercise_id": 1,
"genre": "rock"
}
response = client.post("/recommend/filtered", json=payload)
assert response.status_code == 200
assert "recommendations" in response.json()
|