diff --git a/tests/utils/repository/test_model_repository.py b/tests/utils/repository/test_model_repository.py index 61d1cea..8d7d84a 100644 --- a/tests/utils/repository/test_model_repository.py +++ b/tests/utils/repository/test_model_repository.py @@ -308,6 +308,23 @@ def test_init_artifacts_data_none_y_pred(mock_model_serving_class, mock_logger, repo._init_artifacts_data(mock_train_result) +@patch('model_manager.utils.repository.model_repository.ModelServing') +def test_init_artifacts_data_none_y_train_pred( + mock_model_serving_class, mock_logger, mock_train_result +): + """Test _init_artifacts_data raises ValueError when y_train_pred is None.""" + from model_manager.utils.repository.model_repository import ModelRepository + + repo = ModelRepository( + url='http://mlflow.test', username='user', password='pass', logger=mock_logger + ) + + mock_train_result.y_train_pred = None + + with pytest.raises(ValueError, match='Training predictions .* are None'): + repo._init_artifacts_data(mock_train_result) + + @patch('model_manager.utils.repository.model_repository.ModelServing') @patch('model_manager.utils.repository.model_repository.datetime') @patch('model_manager.utils.repository.model_repository.makedirs')