SIENTIAPDE-1174

SIENTIAPDE-1174 Update tests to enhance MLFlow and workflow activity assertions

- Added logger initialization in MLFlowRepository for improved logging.
- Updated prediction test assertions to reflect changes in output structure.
- Increased activity method call count assertions in workflow tests for accuracy.
This commit is contained in:
vitor-aignosi
2025-08-05 14:19:53 -03:00
parent 3d7de2ef5e
commit fbdf9c3c6c
3 changed files with 8 additions and 5 deletions

View File

@@ -16,7 +16,8 @@ def mlflow_repository():
repo = MLFlowRepository(
host='http://localhost:5000',
username='admin',
password='admin'
password='admin',
logger=MagicMock()
)
return repo
@@ -71,7 +72,9 @@ def test_predict_success(mlflow_repository):
assert output['success'] is True
assert output['content'] == {'prediction': {
0: 3}, 'response_time': ANY}
0: 2,
1: 3
}, 'response_time': ANY}
def test_predict_error(mlflow_repository):