Spaces:
Sleeping
Sleeping
| from fastapi.testclient import TestClient | |
| from api.main import app | |
| client = TestClient(app) | |
| def test_brainstorming(): | |
| payload = { | |
| "preferred_topics": [], | |
| "image_base64_list": [] | |
| } | |
| response = client.post("/api/brainstorm", json=payload) | |
| assert response.status_code == 200 | |
| def test_brainstorming_with_topics(): | |
| payload = { | |
| "preferred_topics": ["fitness","community"], | |
| "image_base64_list": [] | |
| } | |
| response = client.post("/api/brainstorm", json=payload) | |
| assert response.status_code == 200 |