Fine-Tune-LLM / backend /tests /test_config.py
viraj
Initial Commit
dacf14a
raw
history blame
325 Bytes
import pytest
from fastapi.testclient import TestClient
from backend.app.main import app
client = TestClient(app)
def test_configure_model():
response = client.post("/config/", json={"model_name": "distilbert-base-uncased-distilled-squad"})
assert response.status_code == 200
assert "message" in response.json()