/
deniss_shu
/
Project_Lab_7
Обзор
Документация
Войти
/
deniss_shu
/
Project_Lab_7
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
test_model.py
31 строка
1 KB
Shershen-DIXI
update: requirements.txt, test_model.py
03 июн 2026, 15:25
Верифицирован
03 июн 2026, 15:25
2c99edd
Код
Авторство
О чём код?
from unittest.mock import patch, MagicMock def test_output_is_string(): mock_generator = MagicMock(return_value=[{"generated_text": "Привет мир"}]) with patch("transformers.pipeline", return_value=mock_generator): import importlib import model importlib.reload(model) result = model.generate_text("Привет") assert isinstance(result, str) def test_output_not_empty(): mock_generator = MagicMock(return_value=[{"generated_text": "Однажды давно"}]) with patch("transformers.pipeline", return_value=mock_generator): import importlib import model importlib.reload(model) result = model.generate_text("Однажды") assert len(result) > 0 def test_output_starts_with_prompt(): mock_generator = MagicMock(return_value=[{"generated_text": "Москва столица"}]) with patch("transformers.pipeline", return_value=mock_generator): import importlib import model importlib.reload(model) result = model.generate_text("Москва") assert result.startswith("Москва")