File size: 3,173 Bytes
198ccb0 | 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 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 | """Tests for FastAPI application."""
import pytest
from fastapi.testclient import TestClient
from api.main import app
client = TestClient(app)
def test_root():
"""Test root endpoint."""
response = client.get("/")
assert response.status_code == 200
assert "message" in response.json()
def test_health():
"""Test health endpoint."""
response = client.get("/health")
assert response.status_code == 200
data = response.json()
assert "status" in data
assert "model_loaded" in data
def test_classify_invalid_request():
"""Test classification with invalid request."""
# Empty title
response = client.post(
"/classify",
json={"title": ""}
)
assert response.status_code == 422 # Validation error
# Missing title
response = client.post(
"/classify",
json={}
)
assert response.status_code == 422
def test_classify_valid_request():
"""Test classification with valid request."""
# Note: This will fail if model is not loaded
response = client.post(
"/classify",
json={
"title": "Тестовая новость",
"snippet": "Описание новости",
"threshold": 0.5
}
)
# Should either succeed (if model loaded) or return 503 (if not)
assert response.status_code in [200, 503]
def test_classify_batch():
"""Test batch classification."""
response = client.post(
"/classify/batch",
json={
"items": [
{"title": "Новость 1"},
{"title": "Новость 2"}
]
}
)
# Should either succeed or return 503
assert response.status_code in [200, 503]
def test_classify_with_top_k():
"""Test classification with top_k parameter."""
response = client.post(
"/classify",
json={
"title": "Тестовая новость",
"top_k": 5
}
)
assert response.status_code in [200, 503]
if response.status_code == 200:
data = response.json()
assert len(data["predictions"]) <= 5
def test_classify_threshold_validation():
"""Test threshold validation."""
# Invalid threshold (> 1.0)
response = client.post(
"/classify",
json={
"title": "Тест",
"threshold": 1.5
}
)
assert response.status_code == 422
# Invalid threshold (< 0.0)
response = client.post(
"/classify",
json={
"title": "Тест",
"threshold": -0.1
}
)
assert response.status_code == 422
def test_classify_top_k_validation():
"""Test top_k validation."""
# Invalid top_k (> 100)
response = client.post(
"/classify",
json={
"title": "Тест",
"top_k": 200
}
)
assert response.status_code == 422
# Invalid top_k (< 1)
response = client.post(
"/classify",
json={
"title": "Тест",
"top_k": 0
}
)
assert response.status_code == 422
|