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()