tests/test_brainstorm.py

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)
    print(response.json())
    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)
    print(response.json())
    assert response.status_code == 200