1515from src .config .schemas import Enrichment , IntentResult , PredictionResponse
1616from src .utils .normalise import normalise_text
1717
18+
1819# --- This fixture is preserved from your original file ---
1920@pytest .fixture
2021def client () -> Generator [FlaskClient , None , None ]:
@@ -89,7 +90,7 @@ def test_nlu_feedback_success(
8990 # 3. Mock the threshold value
9091 # We need to configure the mock's structure properly
9192 mock_classifier .thresholds = mocker .MagicMock ()
92- mock_classifier .thresholds .sentiment_review_band = 0.75 # Example threshold
93+ mock_classifier .thresholds .sentiment_review_band = 0.75 # Example threshold
9394
9495 # 4. Mock the manifest dictionary (used for model_version)
9596 mock_classifier .manifest = {"version" : "test-v1" }
@@ -106,8 +107,8 @@ def test_nlu_feedback_success(
106107 "model_version" : "test-v1" ,
107108 "parent_label" : "FEEDBACK" ,
108109 "probability" : 0.99 ,
109- "review_status" : "CLASSIFIED" , # Because 0.99 > 0.75
110- "sentiment_label" : "positive" , # Check the extra field
110+ "review_status" : "CLASSIFIED" , # Because 0.99 > 0.75
111+ "sentiment_label" : "positive" , # Check the extra field
111112 }
112113 assert response .json == expected_json
113114
@@ -117,7 +118,7 @@ def test_nlu_feedback_success(
117118 call_args , _ = mock_classifier .sentiment_pipeline .call_args
118119 expected_normalised_text = normalise_text ("the nurses were fantastic" )
119120 assert call_args [0 ] == expected_normalised_text
120-
121+
121122 # Verify 'predict' was NOT called
122123 mock_classifier .predict .assert_not_called ()
123124
0 commit comments