| |
| |
| |
| |
| |
| |
| |
|
|
| from __future__ import annotations |
|
|
| from dataclasses import FrozenInstanceError |
| from pathlib import Path |
|
|
| import pytest |
|
|
| from speech_text_data_aligners.core.models import ( |
| AlignmentFailure, |
| AlignmentOutcome, |
| AlignmentProvenance, |
| AlignmentRequest, |
| AlignmentResult, |
| Diagnostic, |
| DiagnosticSeverity, |
| LexicalWord, |
| WordAlignment, |
| ) |
|
|
|
|
| def test_neutral_records_are_typed_and_immutable() -> None: |
| diagnostic = Diagnostic("note", DiagnosticSeverity.INFO, "safe", "parse") |
| request = AlignmentRequest("r", Path("a.wav"), "Hello", 1.0) |
| word = LexicalWord(0, "Hello", 0, 5) |
| alignment = WordAlignment(word, 0.1, 0.5, 0.9) |
| provenance = AlignmentProvenance( |
| backend_name="fake", artifacts=("ctm",), attributes=(("batch", 1),) |
| ) |
| result = AlignmentResult("r", (alignment,), provenance, (diagnostic,), 0.2) |
| outcome = AlignmentOutcome(result=result) |
|
|
| assert request.transcript == "Hello" |
| assert outcome.succeeded |
| assert outcome.request_id == "r" |
| assert provenance.attributes == (("batch", 1),) |
| with pytest.raises(FrozenInstanceError): |
| word.text = "changed" |
|
|
|
|
| def test_outcome_enforces_exactly_one_variant() -> None: |
| failure = AlignmentFailure("r", "bad", "test", "no") |
| outcome = AlignmentOutcome(failure=failure) |
| assert not outcome.succeeded |
| assert outcome.request_id == "r" |
| with pytest.raises(ValueError, match="exactly one"): |
| AlignmentOutcome() |
| with pytest.raises(ValueError, match="exactly one"): |
| AlignmentOutcome( |
| result=AlignmentResult("r", (), AlignmentProvenance("fake")), |
| failure=failure, |
| ) |
|
|
