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

SIENTIAPDE-1241: Formatted files

Browse Source
This commit is contained in:
Kou-Kinoshita
2025-10-29 11:55:03 -03:00
parent e8250d8dd7
commit 5e763a8a64
3 changed files with 32 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -224,7 +224,9 @@ def test_init_artifacts_data_success(mock_model_serving_class, mock_logger, mock
@patch('model_manager.utils.repository.model_repository.ModelServing')
def test_init_artifacts_data_empty_x_train(mock_model_serving_class, mock_logger, mock_train_result):
def test_init_artifacts_data_empty_x_train(
mock_model_serving_class, mock_logger, mock_train_result
):
"""Test _init_artifacts_data raises ValueError when x_train is empty."""
from model_manager.utils.repository.model_repository import ModelRepository
@@ -239,7 +241,9 @@ def test_init_artifacts_data_empty_x_train(mock_model_serving_class, mock_logger
@patch('model_manager.utils.repository.model_repository.ModelServing')
def test_init_artifacts_data_empty_y_train(mock_model_serving_class, mock_logger, mock_train_result):
def test_init_artifacts_data_empty_y_train(
mock_model_serving_class, mock_logger, mock_train_result
):
"""Test _init_artifacts_data raises ValueError when y_train is empty."""
from model_manager.utils.repository.model_repository import ModelRepository
@@ -331,7 +335,12 @@ def test_setup_run_directory_success(mock_open, mock_copy, mock_model_serving_cl
@patch('model_manager.utils.repository.model_repository.Reports')
@patch('builtins.open', create=True)
def test_generate_report_with_equation(
mock_open, mock_reports_class, mock_json_dump, mock_model_serving_class, mock_logger, mock_train_result
mock_open,
mock_reports_class,
mock_json_dump,
mock_model_serving_class,
mock_logger,
mock_train_result,
):
"""Test _generate_report creates equation JSON artifact."""
from model_manager.utils.repository.model_repository import ModelRepository
@@ -480,7 +489,9 @@ def test_save_run_missing_report(
@patch('model_manager.utils.repository.model_repository.ModelServing')
@patch('model_manager.utils.repository.model_repository.path.exists')
def test_save_run_none_metrics(mock_exists, mock_model_serving_class, mock_logger, mock_train_result):
def test_save_run_none_metrics(
mock_exists, mock_model_serving_class, mock_logger, mock_train_result
):
"""Test _save_run raises ValueError when metrics are None."""
from model_manager.utils.repository.model_repository import ModelRepository
@@ -574,9 +585,7 @@ def test_init_artifacts_data_empty_y_test(mock_model_serving_class, mock_logger,
@patch('model_manager.utils.repository.model_repository.ModelServing')
@patch('model_manager.utils.repository.model_repository.makedirs')
def test_create_run_directory_os_error(
mock_makedirs, mock_model_serving_class, mock_logger
):
def test_create_run_directory_os_error(mock_makedirs, mock_model_serving_class, mock_logger):
"""Test _create_run_directory raises OSError."""
from model_manager.utils.repository.model_repository import ModelRepository
@@ -631,9 +640,7 @@ def test_setup_run_directory_permission_error(
@patch('model_manager.utils.repository.model_repository.ModelServing')
@patch('model_manager.utils.repository.model_repository.shutil.copy')
@patch('builtins.open', create=True)
def test_setup_run_directory_os_error(
mock_open, mock_copy, mock_model_serving_class, mock_logger
):
def test_setup_run_directory_os_error(mock_open, mock_copy, mock_model_serving_class, mock_logger):
"""Test _setup_run_directory raises OSError."""
from model_manager.utils.repository.model_repository import ModelRepository