test(handler): ajout d’un test unitaire pour le modèle few-shot

tests/test_handler.py

@@ -11,6 +11,18 @@ def test_zero_shot_prediction_output():
     print("Résultat brut :", output)
 
     # Vérifie que le format markdown est respecté
+    assert "### Résultat de la classification" in output
+    assert "**toxique**" in output
+    assert "**non-toxique**" in output
+    assert "%" in output
+
+def test_few_shot_prediction_output():
+    from app.handler import predict
+    text = "Tu es un abruti fini"
+    output = predict(text, model_type="few-shot")
+
+    print("Résultat few-shot :", output)
+
     assert "### Résultat de la classification" in output
     assert "**toxique**" in output
     assert "**non-toxique**" in output