/
Dinky6
/
Assessment-Question-Difficulty-Classifier
Обзор
Документация
Войти
/
Dinky6
/
Assessment-Question-Difficulty-Classifier
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
CI/CD
Аналитика
Безопасность
master
tests/test_model.py
43 строки
1 KB
Dinky6
create tests/test_model.py
28 дек 2025, 17:16
28 дек 2025, 17:16
309a123
Код
Авторство
О чём код?
import pytest import numpy as np from sklearn.datasets import make_classification from src.model_training import DifficultyClassifier def test_classifier_initialization(): """Тест инициализации классификатора""" for model_type in ['random_forest', 'svm', 'naive_bayes']: classifier = DifficultyClassifier(model_type) assert classifier.model_type == model_type assert classifier.model is not None def test_invalid_model_type(): """Тест обработки неверного типа модели""" with pytest.raises(ValueError): DifficultyClassifier('invalid_model') def test_training(): """Тест обучения модели""" # Создание тестовых данных X, y = make_classification( n_samples=100, n_features=20, n_classes=3, n_informative=15, random_state=42 ) classifier = DifficultyClassifier('random_forest') classifier.train(X, y) assert hasattr(classifier.model, 'predict') def test_prediction(): """Тест предсказания""" X, y = make_classification( n_samples=100, n_features=20, n_classes=3, n_informative=15, random_state=42 ) classifier = DifficultyClassifier('naive_bayes') classifier.model.fit(X, y) predictions = classifier.predict(X[:5]) assert len(predictions) == 5 assert predictions.shape == (5,)