gitea_admin/sientia-dataops-model-manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1430: Add test case for ValueError when y_train_pred is None during artifact data initialization.

Browse Source
This commit is contained in:
Bruno Domingues
2025-12-18 18:48:42 -03:00
parent 573834fb99
commit 71b654f24c
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tests/utils/repository/test_model_repository.py
View File

@@ -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) 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.ModelServing')
@patch('model_manager.utils.repository.model_repository.datetime') @patch('model_manager.utils.repository.model_repository.datetime')
@patch('model_manager.utils.repository.model_repository.makedirs') @patch('model_manager.utils.repository.model_repository.makedirs')