Initial version of application

tests/test_corrections.py

@@ -0,0 +1,88 @@
+import pytest
+
+from audita.corrections import apply_corrections
+from audita.errors import AuditaValidationError
+from audita.schemas import CorrectionCandidate, parse_transcript_json
+
+
+def _transcript():
+    return parse_transcript_json(
+        """
+        [
+          {"speaker": "Eric", "start": 10.0, "end": 11.0, "text": "I ask Chontia."},
+          {"speaker": "Mike", "start": 0.0, "end": 1.0, "text": "Then Lyra."}
+        ]
+        """
+    )
+
+
+def test_apply_corrections_uses_threshold_and_sorts_chronologically():
+    transcript = _transcript()
+    corrections = [
+        CorrectionCandidate(
+            segment_index=0,
+            speaker="Eric",
+            start=10.0,
+            end=11.0,
+            original_text="I ask Chontia.",
+            corrected_text="I ask Chauntea.",
+            confidence=0.8,
+        )
+    ]
+
+    revised = apply_corrections(transcript, corrections, confidence_threshold=0.8)
+
+    assert [segment.speaker for segment in revised] == ["Mike", "Eric"]
+    assert revised[1].text == "I ask Chauntea."
+
+
+def test_apply_corrections_ignores_below_threshold():
+    transcript = _transcript()
+    corrections = [
+        CorrectionCandidate(
+            segment_index=0,
+            speaker="Eric",
+            start=10.0,
+            end=11.0,
+            original_text="I ask Chontia.",
+            corrected_text="I ask Chauntea.",
+            confidence=0.79,
+        )
+    ]
+
+    revised = apply_corrections(transcript, corrections, confidence_threshold=0.8)
+
+    assert revised[1].text == "I ask Chontia."
+
+
+def test_apply_corrections_rejects_duplicate_targets():
+    transcript = _transcript()
+    correction = CorrectionCandidate(
+        segment_index=0,
+        speaker="Eric",
+        start=10.0,
+        end=11.0,
+        original_text="I ask Chontia.",
+        corrected_text="I ask Chauntea.",
+        confidence=0.8,
+    )
+
+    with pytest.raises(AuditaValidationError):
+        apply_corrections(transcript, [correction, correction], confidence_threshold=0.8)
+
+
+def test_apply_corrections_rejects_mismatched_original_text():
+    transcript = _transcript()
+    correction = CorrectionCandidate(
+        segment_index=0,
+        speaker="Eric",
+        start=10.0,
+        end=11.0,
+        original_text="Different text.",
+        corrected_text="I ask Chauntea.",
+        confidence=0.8,
+    )
+
+    with pytest.raises(AuditaValidationError):
+        apply_corrections(transcript, [correction], confidence_threshold=0.8)
+