SIENTIAPDE-1182

Implement prediction store policy handling in Gates activity

- Added a new method `get_prediction_store_policy` to validate and parse the prediction store policy.
- Updated `format_prediction` method to utilize the new policy handling, allowing for sorting of predictions based on the specified policy.
- Enhanced test coverage for the new policy handling, including various scenarios for valid and invalid policies.
- Removed the obsolete `coverage.sh` script.

tests/laborious/activities/test_gates.py

@@ -322,15 +322,68 @@ async def test_mlflow_content_gate_with_filter(gates_activity):
     gates_activity.send_notification.assert_called()
 
 
+def test_get_prediction_store_policy_invalid_policy(gates_activity):
+    # Arrange
+    prediction_store_policy = 'INVALID_POLICY'
+
+    # Act
+    policy_type, policy_value = gates_activity.get_prediction_store_policy(
+        prediction_store_policy, metadata)
+
+    # Assert
+    assert policy_type == 'lts'
+    assert policy_value == 1
+
+
+def test_get_prediction_store_policy_invalid_policy_value(gates_activity):
+    # Arrange
+    prediction_store_policy = 'abc:INVALID_VALUE'
+
+    # Act
+    policy_type, policy_value = gates_activity.get_prediction_store_policy(
+        prediction_store_policy, metadata)
+
+    # Assert
+    assert policy_type == 'lts'
+    assert policy_value == 1
+
+
+def test_get_prediction_store_policy_valid_policy_type(gates_activity):
+    # Arrange
+    prediction_store_policy = 'abc:1'
+
+    # Act
+    policy_type, policy_value = gates_activity.get_prediction_store_policy(
+        prediction_store_policy, metadata)
+
+    # Assert
+    assert policy_type == 'lts'
+    assert policy_value == 1
+
+
+def test_get_prediction_store_policy_valid_policy(gates_activity):
+    # Arrange
+    prediction_store_policy = 'erl:1'
+
+    # Act
+    policy_type, policy_value = gates_activity.get_prediction_store_policy(
+        prediction_store_policy, metadata)
+
+    # Assert
+    assert policy_type == 'erl'
+    assert policy_value == 1
+
+
 @mark.asyncio
-async def test_format_prediction(gates_activity):
+async def test_format_prediction_no_timestamp(gates_activity):
     # Arrange
     input_data = {
         **metadata,
         'data': {'prediction': [1], 'response_time': [0.1]},
         'timestamp': '2023-05-26 11:12:27',
         'model_id': 'test_model',
-        'prediction_confidence': 0.9
+        'prediction_confidence': 0.9,
+        'prediction_store_policy': 'lts:1'
     }
 
     # Act
@@ -346,6 +399,83 @@ async def test_format_prediction(gates_activity):
     assert result['comments'] == {0: ""}
 
 
+@mark.asyncio
+async def test_format_prediction_with_timestamp_erl(gates_activity):
+    # Arrange
+    input_data = {
+        **metadata,
+        'data': {'prediction': [1, 2, 3],
+                 'response_time': [0.1, 0.2, 0.3],
+                 'timestamp': ['2023-05-26 11:12:27', '2023-05-26 11:12:28', '2023-05-26 11:12:29']},
+        'model_id': 'test_model',
+        'prediction_confidence': 0.9,
+        'prediction_store_policy': 'erl:2'
+    }
+
+    # Act
+    result = await gates_activity.format_prediction(input_data)
+
+    # Assert
+    assert result['prediction'] == {0: 2, 1: 1}
+    assert result['response_time'] == {0: 0.2, 1: 0.1}
+    assert result['timestamp'] == {
+        0: '2023-05-26 11:12:28', 1: '2023-05-26 11:12:27'}
+    assert result['model_id'] == {0: 'test_model', 1: 'test_model'}
+    assert result['prediction_confidence'] == {0: 0.9, 1: 0.9}
+    assert result['prediction_status'] == {0: 'Good', 1: 'Good'}
+    assert result['comments'] == {0: "", 1: ""}
+
+
+@mark.asyncio
+async def test_format_prediction_with_timestamp_lts(gates_activity):
+    # Arrange
+    input_data = {
+        **metadata,
+        'data': {'prediction': [1, 2, 3],
+                 'response_time': [0.1, 0.2, 0.3],
+                 'timestamp': ['2023-05-26 11:12:27', '2023-05-26 11:12:28', '2023-05-26 11:12:29']},
+        'model_id': 'test_model',
+        'prediction_confidence': 0.9,
+        'prediction_store_policy': 'lts:2'
+    }
+
+    # Act
+    result = await gates_activity.format_prediction(input_data)
+
+    # Assert
+    assert result['prediction'] == {0: 3, 1: 2}
+    assert result['response_time'] == {0: 0.3, 1: 0.2}
+    assert result['timestamp'] == {
+        0: '2023-05-26 11:12:29', 1: '2023-05-26 11:12:28'}
+    assert result['model_id'] == {0: 'test_model', 1: 'test_model'}
+    assert result['prediction_confidence'] == {0: 0.9, 1: 0.9}
+    assert result['prediction_status'] == {0: 'Good', 1: 'Good'}
+    assert result['comments'] == {0: "", 1: ""}
+
+
+@mark.asyncio
+async def test_format_prediction_with_timestamp_invalid_policy(gates_activity):
+    # Arrange
+    input_data = {
+        **metadata,
+        'data': {'prediction': [1, 2, 3],
+                 'response_time': [0.1, 0.2, 0.3],
+                 'timestamp': ['2023-05-26 11:12:27', '2023-05-26 11:12:28', '2023-05-26 11:12:29']},
+        'model_id': 'test_model',
+        'prediction_confidence': 0.9,
+        'prediction_store_policy': 'lts:2'
+    }
+    gates_activity.get_prediction_store_policy = MagicMock(
+        return_value=('invalid', 1))
+
+    try:
+        result = await gates_activity.format_prediction(input_data)
+    except ValueError as e:
+        assert str(e) == "Invalid policy type: invalid"
+    else:
+        assert False, "Expected ValueError"
+
+
 @mark.asyncio
 async def test_format_default_prediction(gates_activity):
     # Arrange