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:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user