Dan Walsh
Testing and performance optimisations
124b5b5
from fastapi.testclient import TestClient
import sys
import os
# Import the app from the parent directory
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
from main import app
client = TestClient(app)
def test_summarise_endpoint():
response = client.post(
"/api/summarise",
json={
"text": "This is a test paragraph that should be summarized.",
"max_length": 50,
"min_length": 10
}
)
assert response.status_code == 200
data = response.json()
assert "summary" in data
assert "original_text_length" in data
assert "summary_length" in data