test(handler): ajoute un test pour valider la sortie du modèle fine-tuné

tests/test_handler.py

@@ -24,4 +24,13 @@ def test_few_shot_prediction_output():
     print("Résultat few-shot :", output)
 
     assert "### Résultat de la classification" in output
-    assert "toxique" in output or "non-toxique" in output
+    assert "toxique" in output or "non-toxique" in output
+
+def test_fine_tuned_prediction_output():
+    text = "Tu es stupide"
+    output = predict(text, model_type="fine-tuned")
+
+    print("Résultat fine-tuned :", output)
+
+    assert "### Résultat de la classification" in output
+    assert "toxique" in output or "non-toxique" in output