SIENTIAPDE-994

Add shutdown tests for Activities and OPC classes; improve test assertions
2025-05-28 09:24:16 -03:00
parent b3db8d1190
commit cfb217963b
4 changed files with 60 additions and 8 deletions
--- a/tests/laborious/utils/repository/test_model_repository.py
+++ b/tests/laborious/utils/repository/test_model_repository.py
@@ -251,9 +251,9 @@ def test_predict_success(mlflow_repository):
    mlflow_repository.model_serving.get_cached_predict.assert_called_once_with(
        model_name, data, 1)

-    assert output['success'] == True
+    assert output['success'] is True
    assert output['content'] == {'prediction': {
-        0: 2, 1: 3}, 'response_time': ANY}
+        0: 3}, 'response_time': ANY}


 def test_predict_error(mlflow_repository):