Skip to content

Commit 1cd870f

Browse files
author
Themba Gqaza
committed
lint format
1 parent 8733601 commit 1cd870f

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

tests/test_application.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from src.config.schemas import Enrichment, IntentResult, PredictionResponse
1616
from src.utils.normalise import normalise_text
1717

18+
1819
# --- This fixture is preserved from your original file ---
1920
@pytest.fixture
2021
def 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

Comments
 (0)