File size: 593 Bytes
bbbc13b cfc0db8 bbbc13b cfc0db8 bbbc13b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import json
from fastapi.testclient import TestClient
from megabots import bot, create_api
qnabot = bot("qna-over-docs", index="./examples/files")
app = create_api(qnabot)
client = TestClient(app)
def test_successful_response():
response = client.get("/v1/ask/What is your name?")
assert response.status_code == 200
assert "answer" in response.json()
assert isinstance(response.json()["answer"], str)
def test_missing_question_parameter():
response = client.get("/v1/ask/")
assert response.status_code == 404
assert response.json() == {"detail": "Not Found"}
|