Spaces:
Sleeping
Sleeping
File size: 549 Bytes
4a45dbd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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 |