SIENTIAPDE-1182

SIENTIAPDE-1205: Enhance predictions_batch function and update test cases

- Added 'predictions_storage_policy' to the predictions_batch function for improved configuration handling.
- Updated test cases to include 'predictions_storage_policy' in expected results, ensuring consistency in testing.
This commit is contained in:
vitor-aignosi
2025-08-28 16:32:06 -03:00
parent d701943586
commit 36f1388301
2 changed files with 6 additions and 3 deletions

View File

@@ -197,7 +197,8 @@ def test_predictions_batch(mock_process_path_priority,
}
],
"path_priority": ["STOP", "CONTINUE", "REPEAT"],
"datetime_columns": ["timestamp"]
"datetime_columns": ["timestamp"],
"predictions_storage_policy": "erl:1"
}
result = predictions_batch(config)
@@ -272,7 +273,8 @@ def test_predictions_batch(mock_process_path_priority,
}
},
"path_priority": ["STOP", "CONTINUE", "REPEAT"],
"datetime_columns": ["timestamp"]
"datetime_columns": ["timestamp"],
"predictions_storage_policy": "erl:1"
}
assert result == expected