Raheel Abdul Rehman
Initial Push
c31d1ca
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()