/
vnss64657
/
SE_FastAPI_Example
Обзор
Документация
Войти
/
vnss64657
/
SE_FastAPI_Example
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
main
tests/test_main.py
33 строки
1012 B
Вологин Никита Сергеевич
Fix flake8 warnings: no newline at end of file
11 май 2026, 18:48
11 май 2026, 18:48
61f6288
Код
Авторство
О чём код?
from fastapi.testclient import TestClient from unittest.mock import patch, MagicMock from main import app client = TestClient(app) def test_root(): response = client.get("/") assert response.status_code == 200 assert response.json() == {"message": "FastAPI service started!"} @patch("main.get_classifier") def test_predict(mock_get_classifier): # Создаём мок-классификатор mock_classifier = MagicMock() mock_classifier.return_value = [{"label": "POSITIVE", "score": 0.99}] mock_get_classifier.return_value = mock_classifier response = client.post("/predict/", json={"text": "Hello"}) assert response.status_code == 200 data = response.json() assert len(data) == 1 assert data[0]["label"] == "POSITIVE" assert data[0]["score"] == 0.99 mock_get_classifier.assert_called_once() def test_predict_empty_text(): response = client.post("/predict/", json={"text": ""}) assert response.status_code == 422 # Validation error