File size: 2,104 Bytes
638c806 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 | from __future__ import annotations
from fastapi.testclient import TestClient
from promptboost.main import app
def test_health_route(temp_db) -> None:
client = TestClient(app)
response = client.get("/health")
assert response.status_code == 200
assert response.json()["ok"] is True
def test_index_route(temp_db) -> None:
client = TestClient(app)
response = client.get("/")
assert response.status_code == 200
assert "PromptBoost" in response.text
def test_boost_and_runs_routes(temp_db) -> None:
client = TestClient(app)
response = client.post(
"/boost",
json={
"raw_prompt": "Fix the API bug and add tests without unrelated refactors.",
"target_harness": "codex",
"target_model": "gpt",
},
)
assert response.status_code == 200
body = response.json()
assert body["result"]["detected_task_type"] == "coding_patch"
assert body["result"]["input_adequacy"]["status"] in {"adequate", "needs_clarification"}
assert "no_new_facts_result" in body["result"]
assert body["result"]["no_new_facts_passed"] is True
assert body["run"]["id"] == 1
runs = client.get("/api/runs").json()["runs"]
assert len(runs) == 1
def test_promptgarage_samples_route(temp_db) -> None:
client = TestClient(app)
response = client.get("/api/promptgarage/samples?limit=2")
assert response.status_code == 200
assert "samples" in response.json()
def test_matrix_lab_page_and_deterministic_route(temp_db) -> None:
client = TestClient(app)
page = client.get("/matrix-lab")
assert page.status_code == 200
assert "Matrix Lab" in page.text
response = client.post(
"/api/matrix/deterministic",
json={
"raw_prompt": "Fix the API route, add tests, and avoid unrelated refactors.",
},
)
assert response.status_code == 200
body = response.json()
assert len(body["harnesses"]) == 10
assert len(body["models"]) == 10
assert len(body["rows"]) == 100
assert body["rows"][0]["metrics"]["words"] > 0
|