Spaces:
Sleeping
Sleeping
| 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() | |