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

@@ -79,7 +79,7 @@ async def test_run_none_path_flag(workflow_mock, format_and_export_prediction):
start_to_close_timeout=ANY
)])
assert workflow_mock.execute_activity_method.call_count == 2
assert workflow_mock.execute_activity_method.call_count == 3
assert workflow_mock.execute_local_activity_method.call_count == 1
@@ -145,5 +145,5 @@ async def test_run_default_path_flag(workflow_mock, format_and_export_prediction
)
])
assert workflow_mock.execute_activity_method.call_count == 2
assert workflow_mock.execute_activity_method.call_count == 3
assert workflow_mock.execute_local_activity_method.call_count == 1