dismissed wrong label triggers recorrection

tests/test_prompting.py

@@ -7,6 +7,21 @@ from copienator import prompting
 
 
 class PromptingTests(unittest.TestCase):
+    def test_confirmed_label_request_keeps_grading_material_and_identifies_single_answer(self):
+        with (
+            patch.object(prompting, "get_single_image_bytes", return_value=b"answer"),
+            patch.object(prompting, "get_label_text_content", return_value="Question"),
+            patch.object(prompting, "get_label_sol_content", return_value="Solution"),
+            patch.object(prompting, "get_label_persp_content", return_value="Marking criteria"),
+        ):
+            contents, config = prompting.request_for_confirmed_label("evaluation", "answer.pdf", "Ex 2 : 2)b)", "01")
+        prompt = contents[0].parts[1].text
+        self.assertEqual(contents[0].parts[0].inline_data.data, b"answer")
+        for text in ("Question", "Solution", "Marking criteria", "`01`", "`Ex 2 : 2)b)`",
+                     "Reprends sa correction mathématique", "Ne renvoie pas `wrong-label`"):
+            self.assertIn(text, prompt)
+        self.assertEqual(config.response_mime_type, "application/json")
+
     def test_live_prompt_includes_only_metadata_ids_in_order(self) -> None:
         with tempfile.TemporaryDirectory() as directory:
             image = Path(directory) / "Group_16.jpg"