tests/test_image_generation.py

from fastapi.testclient import TestClient
from api.main import app

client = TestClient(app)

def test_image_generation():
    response = client.get("/api/generate-image")
    print(response.json())
    assert response.status_code == 200