Spaces:
Sleeping
Sleeping
File size: 325 Bytes
dacf14a |
1 2 3 4 5 6 7 8 9 10 |
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() |